{
 "cells": [
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Utilizing Custom ONNX Models Stored in Hugging Face within HSSM\n",
    "This guide demonstrates how to use a custom ONNX model stored in Hugging Face to generate a log-likelihood in HSSM."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Colab Instructions\n",
    "\n",
    "If you would like to run this tutorial on Google colab, please click this [link](https://github.com/lnccbrown/HSSM/blob/main/docs/tutorials/hugging_face_onnx_models.ipynb). \n",
    "\n",
    "Once you are *in the colab*, follow the *installation instructions below* and then **restart your runtime**. \n",
    "\n",
    "Just **uncomment the code in the next code cell** and run it!\n",
    "\n",
    "**NOTE**:\n",
    "\n",
    "You may want to *switch your runtime* to have a GPU or TPU. To do so, go to *Runtime* > *Change runtime type* and select the desired hardware accelerator.\n",
    "\n",
    "Note that if you switch your runtime you have to follow the installation instructions again."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# !pip install numpy==1.23.4\n",
    "# !pip install git+https://github.com/lnccbrown/hssm@main\n",
    "# !pip install git+https://github.com/brown-ccv/hddm-wfpt@main\n",
    "# !pip install numpyro"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Load Modules"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import arviz as az\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "import pytensor\n",
    "\n",
    "import hssm\n",
    "import ssms.basic_simulators\n",
    "\n",
    "pytensor.config.floatX = \"float32\""
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Simulating Data\n",
    "Start by simulating some data:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>rt</th>\n",
       "      <th>response</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.367000</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1.430997</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0.424000</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1.593004</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0.560000</td>\n",
       "      <td>-1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>995</th>\n",
       "      <td>1.707010</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>996</th>\n",
       "      <td>0.654999</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>997</th>\n",
       "      <td>1.201992</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>998</th>\n",
       "      <td>4.502878</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>999</th>\n",
       "      <td>2.203033</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>1000 rows × 2 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "           rt  response\n",
       "0    0.367000       1.0\n",
       "1    1.430997       1.0\n",
       "2    0.424000       1.0\n",
       "3    1.593004       1.0\n",
       "4    0.560000      -1.0\n",
       "..        ...       ...\n",
       "995  1.707010       1.0\n",
       "996  0.654999       1.0\n",
       "997  1.201992       1.0\n",
       "998  4.502878       1.0\n",
       "999  2.203033       1.0\n",
       "\n",
       "[1000 rows x 2 columns]"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Simulate some data\n",
    "v_true, a_true, z_true, alpha_true, t_true = [0.5, 1.5, 0.5, 0.5, 0.3]\n",
    "obs_angle = ssms.basic_simulators.simulator(\n",
    "    [v_true, a_true, z_true, alpha_true, t_true], model=\"levy\", n_samples=1000\n",
    ")\n",
    "obs_angle = np.column_stack([obs_angle[\"rts\"][:, 0], obs_angle[\"choices\"][:, 0]])\n",
    "dataset_lan = pd.DataFrame(obs_angle, columns=[\"rt\", \"response\"])\n",
    "dataset_lan"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Loading ONNX Model from Hugging Face Repository. \n",
    "If `lexy.onnx` stored in [`hugging face repository`](https://huggingface.co/franklab/HSSM), you can specify it in HSSM model as shown below:. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [],
   "source": [
    "my_hssm = hssm.HSSM(\n",
    "    data=dataset_lan,\n",
    "    model=\"custom\",\n",
    "    model_config={\n",
    "        \"list_params\": [\"v\", \"a\", \"z\", \"alpha\", \"t\"],\n",
    "        \"backend\": \"jax\",\n",
    "        \"bounds\": {\n",
    "            \"v\": (-3.0, 3.0),\n",
    "            \"a\": (0.3, 3.0),\n",
    "            \"z\": (0.1, 0.9),\n",
    "            \"alpha\": (1.0, 2.0),\n",
    "            \"t\": (1e-3, 2.0),\n",
    "        },\n",
    "    },\n",
    "    loglik_kind=\"approx_differentiable\",\n",
    "    loglik=\"angle.onnx\",\n",
    ")"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "This creates an HSSM object my_hssm using the custom ONNX model levy.onnx from the Hugging Face repository."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Auto-assigning NUTS sampler...\n",
      "Initializing NUTS using jitter+adapt_diag...\n",
      "Multiprocess sampling (2 chains in 2 jobs)\n",
      "NUTS: [alpha, z, t, a, v]\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "\n",
       "<style>\n",
       "    /* Turns off some styling */\n",
       "    progress {\n",
       "        /* gets rid of default border in Firefox and Opera. */\n",
       "        border: none;\n",
       "        /* Needs to be in here for Safari polyfill so background images work as expected. */\n",
       "        background-size: auto;\n",
       "    }\n",
       "    progress:not([value]), progress:not([value])::-webkit-progress-bar {\n",
       "        background: repeating-linear-gradient(45deg, #7e7e7e, #7e7e7e 10px, #5c5c5c 10px, #5c5c5c 20px);\n",
       "    }\n",
       "    .progress-bar-interrupted, .progress-bar-interrupted::-webkit-progress-bar {\n",
       "        background: #F44336;\n",
       "    }\n",
       "</style>\n"
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/html": [
       "\n",
       "    <div>\n",
       "      <progress value='2000' class='' max='2000' style='width:300px; height:20px; vertical-align: middle;'></progress>\n",
       "      100.00% [2000/2000 00:24&lt;00:00 Sampling 2 chains, 0 divergences]\n",
       "    </div>\n",
       "    "
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Sampling 2 chains for 500 tune and 500 draw iterations (1_000 + 1_000 draws total) took 35 seconds.\n",
      "We recommend running at least 4 chains for robust computation of convergence diagnostics\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "\n",
       "            <div>\n",
       "              <div class='xr-header'>\n",
       "                <div class=\"xr-obj-type\">arviz.InferenceData</div>\n",
       "              </div>\n",
       "              <ul class=\"xr-sections group-sections\">\n",
       "              \n",
       "            <li class = \"xr-section-item\">\n",
       "                  <input id=\"idata_posterior4274c49e-74a6-4da8-8a61-b6a7482fcfe0\" class=\"xr-section-summary-in\" type=\"checkbox\">\n",
       "                  <label for=\"idata_posterior4274c49e-74a6-4da8-8a61-b6a7482fcfe0\" class = \"xr-section-summary\">posterior</label>\n",
       "                  <div class=\"xr-section-inline-details\"></div>\n",
       "                  <div class=\"xr-section-details\">\n",
       "                      <ul id=\"xr-dataset-coord-list\" class=\"xr-var-list\">\n",
       "                          <div style=\"padding-left:2rem;\"><div><svg style=\"position: absolute; width: 0; height: 0; overflow: hidden\">\n",
       "<defs>\n",
       "<symbol id=\"icon-database\" viewBox=\"0 0 32 32\">\n",
       "<path d=\"M16 0c-8.837 0-16 2.239-16 5v4c0 2.761 7.163 5 16 5s16-2.239 16-5v-4c0-2.761-7.163-5-16-5z\"></path>\n",
       "<path d=\"M16 17c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n",
       "<path d=\"M16 26c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n",
       "</symbol>\n",
       "<symbol id=\"icon-file-text2\" viewBox=\"0 0 32 32\">\n",
       "<path d=\"M28.681 7.159c-0.694-0.947-1.662-2.053-2.724-3.116s-2.169-2.030-3.116-2.724c-1.612-1.182-2.393-1.319-2.841-1.319h-15.5c-1.378 0-2.5 1.121-2.5 2.5v27c0 1.378 1.122 2.5 2.5 2.5h23c1.378 0 2.5-1.122 2.5-2.5v-19.5c0-0.448-0.137-1.23-1.319-2.841zM24.543 5.457c0.959 0.959 1.712 1.825 2.268 2.543h-4.811v-4.811c0.718 0.556 1.584 1.309 2.543 2.268zM28 29.5c0 0.271-0.229 0.5-0.5 0.5h-23c-0.271 0-0.5-0.229-0.5-0.5v-27c0-0.271 0.229-0.5 0.5-0.5 0 0 15.499-0 15.5 0v7c0 0.552 0.448 1 1 1h7v19.5z\"></path>\n",
       "<path d=\"M23 26h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "<path d=\"M23 22h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "<path d=\"M23 18h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "</symbol>\n",
       "</defs>\n",
       "</svg>\n",
       "<style>/* CSS stylesheet for displaying xarray objects in jupyterlab.\n",
       " *\n",
       " */\n",
       "\n",
       ":root {\n",
       "  --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));\n",
       "  --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));\n",
       "  --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));\n",
       "  --xr-border-color: var(--jp-border-color2, #e0e0e0);\n",
       "  --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);\n",
       "  --xr-background-color: var(--jp-layout-color0, white);\n",
       "  --xr-background-color-row-even: var(--jp-layout-color1, white);\n",
       "  --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);\n",
       "}\n",
       "\n",
       "html[theme=dark],\n",
       "body[data-theme=dark],\n",
       "body.vscode-dark {\n",
       "  --xr-font-color0: rgba(255, 255, 255, 1);\n",
       "  --xr-font-color2: rgba(255, 255, 255, 0.54);\n",
       "  --xr-font-color3: rgba(255, 255, 255, 0.38);\n",
       "  --xr-border-color: #1F1F1F;\n",
       "  --xr-disabled-color: #515151;\n",
       "  --xr-background-color: #111111;\n",
       "  --xr-background-color-row-even: #111111;\n",
       "  --xr-background-color-row-odd: #313131;\n",
       "}\n",
       "\n",
       ".xr-wrap {\n",
       "  display: block !important;\n",
       "  min-width: 300px;\n",
       "  max-width: 700px;\n",
       "}\n",
       "\n",
       ".xr-text-repr-fallback {\n",
       "  /* fallback to plain text repr when CSS is not injected (untrusted notebook) */\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-header {\n",
       "  padding-top: 6px;\n",
       "  padding-bottom: 6px;\n",
       "  margin-bottom: 4px;\n",
       "  border-bottom: solid 1px var(--xr-border-color);\n",
       "}\n",
       "\n",
       ".xr-header > div,\n",
       ".xr-header > ul {\n",
       "  display: inline;\n",
       "  margin-top: 0;\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-obj-type,\n",
       ".xr-array-name {\n",
       "  margin-left: 2px;\n",
       "  margin-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-obj-type {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-sections {\n",
       "  padding-left: 0 !important;\n",
       "  display: grid;\n",
       "  grid-template-columns: 150px auto auto 1fr 20px 20px;\n",
       "}\n",
       "\n",
       ".xr-section-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-section-item input {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-item input + label {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label {\n",
       "  cursor: pointer;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label:hover {\n",
       "  color: var(--xr-font-color0);\n",
       "}\n",
       "\n",
       ".xr-section-summary {\n",
       "  grid-column: 1;\n",
       "  color: var(--xr-font-color2);\n",
       "  font-weight: 500;\n",
       "}\n",
       "\n",
       ".xr-section-summary > span {\n",
       "  display: inline-block;\n",
       "  padding-left: 0.5em;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in + label:before {\n",
       "  display: inline-block;\n",
       "  content: '►';\n",
       "  font-size: 11px;\n",
       "  width: 15px;\n",
       "  text-align: center;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label:before {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label:before {\n",
       "  content: '▼';\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label > span {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-summary,\n",
       ".xr-section-inline-details {\n",
       "  padding-top: 4px;\n",
       "  padding-bottom: 4px;\n",
       "}\n",
       "\n",
       ".xr-section-inline-details {\n",
       "  grid-column: 2 / -1;\n",
       "}\n",
       "\n",
       ".xr-section-details {\n",
       "  display: none;\n",
       "  grid-column: 1 / -1;\n",
       "  margin-bottom: 5px;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked ~ .xr-section-details {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-array-wrap {\n",
       "  grid-column: 1 / -1;\n",
       "  display: grid;\n",
       "  grid-template-columns: 20px auto;\n",
       "}\n",
       "\n",
       ".xr-array-wrap > label {\n",
       "  grid-column: 1;\n",
       "  vertical-align: top;\n",
       "}\n",
       "\n",
       ".xr-preview {\n",
       "  color: var(--xr-font-color3);\n",
       "}\n",
       "\n",
       ".xr-array-preview,\n",
       ".xr-array-data {\n",
       "  padding: 0 5px !important;\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-array-data,\n",
       ".xr-array-in:checked ~ .xr-array-preview {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-array-in:checked ~ .xr-array-data,\n",
       ".xr-array-preview {\n",
       "  display: inline-block;\n",
       "}\n",
       "\n",
       ".xr-dim-list {\n",
       "  display: inline-block !important;\n",
       "  list-style: none;\n",
       "  padding: 0 !important;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list li {\n",
       "  display: inline-block;\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list:before {\n",
       "  content: '(';\n",
       "}\n",
       "\n",
       ".xr-dim-list:after {\n",
       "  content: ')';\n",
       "}\n",
       "\n",
       ".xr-dim-list li:not(:last-child):after {\n",
       "  content: ',';\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-has-index {\n",
       "  font-weight: bold;\n",
       "}\n",
       "\n",
       ".xr-var-list,\n",
       ".xr-var-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-var-item > div,\n",
       ".xr-var-item label,\n",
       ".xr-var-item > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-even);\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-var-item > .xr-var-name:hover span {\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-var-list > li:nth-child(odd) > div,\n",
       ".xr-var-list > li:nth-child(odd) > label,\n",
       ".xr-var-list > li:nth-child(odd) > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-odd);\n",
       "}\n",
       "\n",
       ".xr-var-name {\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-var-dims {\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-var-dtype {\n",
       "  grid-column: 3;\n",
       "  text-align: right;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-preview {\n",
       "  grid-column: 4;\n",
       "}\n",
       "\n",
       ".xr-index-preview {\n",
       "  grid-column: 2 / 5;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-name,\n",
       ".xr-var-dims,\n",
       ".xr-var-dtype,\n",
       ".xr-preview,\n",
       ".xr-attrs dt {\n",
       "  white-space: nowrap;\n",
       "  overflow: hidden;\n",
       "  text-overflow: ellipsis;\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-var-name:hover,\n",
       ".xr-var-dims:hover,\n",
       ".xr-var-dtype:hover,\n",
       ".xr-attrs dt:hover {\n",
       "  overflow: visible;\n",
       "  width: auto;\n",
       "  z-index: 1;\n",
       "}\n",
       "\n",
       ".xr-var-attrs,\n",
       ".xr-var-data,\n",
       ".xr-index-data {\n",
       "  display: none;\n",
       "  background-color: var(--xr-background-color) !important;\n",
       "  padding-bottom: 5px !important;\n",
       "}\n",
       "\n",
       ".xr-var-attrs-in:checked ~ .xr-var-attrs,\n",
       ".xr-var-data-in:checked ~ .xr-var-data,\n",
       ".xr-index-data-in:checked ~ .xr-index-data {\n",
       "  display: block;\n",
       "}\n",
       "\n",
       ".xr-var-data > table {\n",
       "  float: right;\n",
       "}\n",
       "\n",
       ".xr-var-name span,\n",
       ".xr-var-data,\n",
       ".xr-index-name div,\n",
       ".xr-index-data,\n",
       ".xr-attrs {\n",
       "  padding-left: 25px !important;\n",
       "}\n",
       "\n",
       ".xr-attrs,\n",
       ".xr-var-attrs,\n",
       ".xr-var-data,\n",
       ".xr-index-data {\n",
       "  grid-column: 1 / -1;\n",
       "}\n",
       "\n",
       "dl.xr-attrs {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  display: grid;\n",
       "  grid-template-columns: 125px auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt,\n",
       ".xr-attrs dd {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  float: left;\n",
       "  padding-right: 10px;\n",
       "  width: auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt {\n",
       "  font-weight: normal;\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-attrs dt:hover span {\n",
       "  display: inline-block;\n",
       "  background: var(--xr-background-color);\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-attrs dd {\n",
       "  grid-column: 2;\n",
       "  white-space: pre-wrap;\n",
       "  word-break: break-all;\n",
       "}\n",
       "\n",
       ".xr-icon-database,\n",
       ".xr-icon-file-text2,\n",
       ".xr-no-icon {\n",
       "  display: inline-block;\n",
       "  vertical-align: middle;\n",
       "  width: 1em;\n",
       "  height: 1.5em !important;\n",
       "  stroke-width: 0;\n",
       "  stroke: currentColor;\n",
       "  fill: currentColor;\n",
       "}\n",
       "</style><pre class='xr-text-repr-fallback'>&lt;xarray.Dataset&gt;\n",
       "Dimensions:  (chain: 2, draw: 500)\n",
       "Coordinates:\n",
       "  * chain    (chain) int64 0 1\n",
       "  * draw     (draw) int64 0 1 2 3 4 5 6 7 8 ... 492 493 494 495 496 497 498 499\n",
       "Data variables:\n",
       "    alpha    (chain, draw) float32 1.005 1.021 1.015 1.05 ... 1.006 1.044 1.001\n",
       "    z        (chain, draw) float32 0.4778 0.4628 0.4369 ... 0.4857 0.451 0.4947\n",
       "    t        (chain, draw) float32 0.3074 0.3009 0.3084 ... 0.3078 0.3148 0.2974\n",
       "    a        (chain, draw) float32 1.246 1.253 1.286 1.251 ... 1.246 1.244 1.249\n",
       "    v        (chain, draw) float32 0.3329 0.3736 0.4343 ... 0.3108 0.4092 0.3667\n",
       "Attributes:\n",
       "    created_at:                  2023-06-22T18:12:24.938346\n",
       "    arviz_version:               0.14.0\n",
       "    inference_library:           pymc\n",
       "    inference_library_version:   5.5.0\n",
       "    sampling_time:               35.316415786743164\n",
       "    tuning_steps:                500\n",
       "    modeling_interface:          bambi\n",
       "    modeling_interface_version:  0.11.0</pre><div class='xr-wrap' style='display:none'><div class='xr-header'><div class='xr-obj-type'>xarray.Dataset</div></div><ul class='xr-sections'><li class='xr-section-item'><input id='section-8552b9fa-0a8a-4624-8735-aae86db944f4' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-8552b9fa-0a8a-4624-8735-aae86db944f4' class='xr-section-summary'  title='Expand/collapse section'>Dimensions:</label><div class='xr-section-inline-details'><ul class='xr-dim-list'><li><span class='xr-has-index'>chain</span>: 2</li><li><span class='xr-has-index'>draw</span>: 500</li></ul></div><div class='xr-section-details'></div></li><li class='xr-section-item'><input id='section-a70d7d41-808f-4b46-8812-437287bab43e' class='xr-section-summary-in' type='checkbox'  checked><label for='section-a70d7d41-808f-4b46-8812-437287bab43e' class='xr-section-summary' >Coordinates: <span>(2)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span class='xr-has-index'>chain</span></div><div class='xr-var-dims'>(chain)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>0 1</div><input id='attrs-5b447d75-49c7-4c4f-a279-096e45e89770' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-5b447d75-49c7-4c4f-a279-096e45e89770' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-e6cc4dbb-9130-4048-ab99-354b885c3be4' class='xr-var-data-in' type='checkbox'><label for='data-e6cc4dbb-9130-4048-ab99-354b885c3be4' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([0, 1])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span class='xr-has-index'>draw</span></div><div class='xr-var-dims'>(draw)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>0 1 2 3 4 5 ... 495 496 497 498 499</div><input id='attrs-c41d2184-d617-4657-9216-d293d5fd254e' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-c41d2184-d617-4657-9216-d293d5fd254e' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-d0b78d8f-0668-4cd9-8ecf-6b3d338c4fb7' class='xr-var-data-in' type='checkbox'><label for='data-d0b78d8f-0668-4cd9-8ecf-6b3d338c4fb7' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([  0,   1,   2, ..., 497, 498, 499])</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-14c4fe55-6245-4321-ac2b-84438047fef0' class='xr-section-summary-in' type='checkbox'  checked><label for='section-14c4fe55-6245-4321-ac2b-84438047fef0' class='xr-section-summary' >Data variables: <span>(5)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span>alpha</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float32</div><div class='xr-var-preview xr-preview'>1.005 1.021 1.015 ... 1.044 1.001</div><input id='attrs-9efaecd1-354f-4d3f-a294-bb35d22781d3' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-9efaecd1-354f-4d3f-a294-bb35d22781d3' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-3c22617e-3206-447b-9b26-1c21f37d4cbb' class='xr-var-data-in' type='checkbox'><label for='data-3c22617e-3206-447b-9b26-1c21f37d4cbb' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[1.0049396, 1.0210133, 1.0150042, 1.0503689, 1.0159004, 1.0527126,\n",
       "        1.0104135, 1.0520232, 1.034015 , 1.020892 , 1.0103153, 1.09737  ,\n",
       "        1.000638 , 1.002539 , 1.0138824, 1.0096576, 1.1605872, 1.0001441,\n",
       "        1.000229 , 1.000072 , 1.0108839, 1.011656 , 1.0105702, 1.0734832,\n",
       "        1.0065091, 1.0802119, 1.0022935, 1.0070705, 1.0021468, 1.0059818,\n",
       "        1.0343146, 1.0155642, 1.005657 , 1.0170338, 1.0209619, 1.0097249,\n",
       "        1.017627 , 1.0289446, 1.0121984, 1.011126 , 1.0302304, 1.0075248,\n",
       "        1.0079168, 1.0357382, 1.0105025, 1.0598809, 1.023328 , 1.0157487,\n",
       "        1.0298902, 1.0274063, 1.0236958, 1.0264926, 1.0277157, 1.0133216,\n",
       "        1.0552391, 1.0132465, 1.010351 , 1.0082686, 1.0197146, 1.0324031,\n",
       "        1.0261567, 1.016435 , 1.0080427, 1.0080427, 1.0154891, 1.0206103,\n",
       "        1.0360273, 1.0007471, 1.0275594, 1.0229585, 1.0394834, 1.0279192,\n",
       "        1.013108 , 1.0360042, 1.0210186, 1.0052209, 1.0005298, 1.0038725,\n",
       "        1.0400771, 1.0139259, 1.0372728, 1.0372728, 1.0071579, 1.0026712,\n",
       "        1.0360725, 1.0180074, 1.017007 , 1.0212777, 1.0275009, 1.0153171,\n",
       "        1.0128216, 1.0304269, 1.0106446, 1.0136783, 1.0334421, 1.005844 ,\n",
       "        1.0209565, 1.0383129, 1.0186288, 1.0182142, 1.029469 , 1.0267245,\n",
       "        1.0836058, 1.0048811, 1.0669197, 1.0085297, 1.0236428, 1.040352 ,\n",
       "        1.0026131, 1.0105039, 1.0209227, 1.0161304, 1.0306715, 1.0922544,\n",
       "        1.0003618, 1.0470183, 1.0558591, 1.0056723, 1.002141 , 1.0010923,\n",
       "...\n",
       "        1.0021063, 1.038385 , 1.0140891, 1.0188665, 1.0232716, 1.0063641,\n",
       "        1.0052445, 1.0495144, 1.0199788, 1.0081484, 1.03733  , 1.0190879,\n",
       "        1.0265719, 1.002509 , 1.002509 , 1.006785 , 1.0120864, 1.02432  ,\n",
       "        1.0122241, 1.0154403, 1.0494252, 1.022783 , 1.0283655, 1.0234962,\n",
       "        1.0220718, 1.010717 , 1.0537355, 1.0513532, 1.062371 , 1.0005125,\n",
       "        1.0112431, 1.0779394, 1.0222737, 1.0343578, 1.0039898, 1.0619081,\n",
       "        1.0170121, 1.0010877, 1.0361071, 1.0735376, 1.0187467, 1.0019042,\n",
       "        1.0046339, 1.0626607, 1.0455712, 1.025816 , 1.0266417, 1.0184124,\n",
       "        1.0090089, 1.0445536, 1.0112643, 1.0059764, 1.0966759, 1.000895 ,\n",
       "        1.0008137, 1.0161102, 1.102823 , 1.0023849, 1.0098819, 1.0019971,\n",
       "        1.0092238, 1.0765254, 1.0765254, 1.0018564, 1.0312988, 1.0116545,\n",
       "        1.0417817, 1.0133657, 1.0360593, 1.0175686, 1.0135255, 1.0179013,\n",
       "        1.0367948, 1.0099232, 1.0063677, 1.0053743, 1.0705166, 1.0081453,\n",
       "        1.0059935, 1.050851 , 1.0157342, 1.0323342, 1.0018616, 1.0118489,\n",
       "        1.0380523, 1.0256711, 1.0108318, 1.0404545, 1.0235987, 1.0239229,\n",
       "        1.0310103, 1.0139579, 1.0142511, 1.0265923, 1.0088834, 1.0094961,\n",
       "        1.0144334, 1.0010761, 1.0438523, 1.0127017, 1.0173893, 1.0340981,\n",
       "        1.0608739, 1.0329942, 1.0095947, 1.0198166, 1.0228288, 1.0088893,\n",
       "        1.0157295, 1.0389754, 1.0156997, 1.0069684, 1.0135572, 1.0058519,\n",
       "        1.0442272, 1.0008465]], dtype=float32)</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>z</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float32</div><div class='xr-var-preview xr-preview'>0.4778 0.4628 ... 0.451 0.4947</div><input id='attrs-db7e9930-19ea-4546-9c29-bd5207eaf358' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-db7e9930-19ea-4546-9c29-bd5207eaf358' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-0f751bb2-0638-4f18-9bfb-60236aba6789' class='xr-var-data-in' type='checkbox'><label for='data-0f751bb2-0638-4f18-9bfb-60236aba6789' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.47779563, 0.4627997 , 0.43686908, 0.46775904, 0.461738  ,\n",
       "        0.43930894, 0.48600417, 0.45804584, 0.47959134, 0.41300634,\n",
       "        0.44678587, 0.4464887 , 0.43444556, 0.43889046, 0.44524664,\n",
       "        0.46943778, 0.46775895, 0.42941922, 0.43050334, 0.4520984 ,\n",
       "        0.45481548, 0.41737664, 0.4435914 , 0.42872393, 0.47540691,\n",
       "        0.41004908, 0.47947836, 0.45461667, 0.47919196, 0.4436096 ,\n",
       "        0.45479944, 0.39966708, 0.43752933, 0.4255519 , 0.48758432,\n",
       "        0.40975812, 0.4924298 , 0.39970836, 0.4308459 , 0.44680712,\n",
       "        0.4480644 , 0.47089934, 0.4414161 , 0.42451882, 0.4373785 ,\n",
       "        0.43271518, 0.44313774, 0.41538334, 0.41651404, 0.4112932 ,\n",
       "        0.48470643, 0.4486939 , 0.44964033, 0.39621702, 0.4129896 ,\n",
       "        0.493085  , 0.44181818, 0.45356107, 0.48087886, 0.45241565,\n",
       "        0.44998577, 0.4813748 , 0.47181526, 0.47181526, 0.4713124 ,\n",
       "        0.49733207, 0.4275319 , 0.510853  , 0.41376194, 0.44647372,\n",
       "        0.44213808, 0.44093975, 0.40662268, 0.45623168, 0.44994262,\n",
       "        0.46639568, 0.45664462, 0.4759493 , 0.43108153, 0.42105362,\n",
       "        0.432979  , 0.432979  , 0.44521737, 0.4447575 , 0.46482587,\n",
       "        0.46926793, 0.45863396, 0.47445345, 0.41435516, 0.4123544 ,\n",
       "        0.47431886, 0.4490975 , 0.45803416, 0.4644014 , 0.43658593,\n",
       "        0.4475816 , 0.4866408 , 0.43765748, 0.46942818, 0.45191085,\n",
       "...\n",
       "        0.42715698, 0.4722904 , 0.4955282 , 0.45233482, 0.42898786,\n",
       "        0.41926324, 0.38091078, 0.4425263 , 0.4597411 , 0.47722977,\n",
       "        0.46672413, 0.47268838, 0.49460113, 0.46823293, 0.39926824,\n",
       "        0.44187194, 0.4472052 , 0.45369294, 0.44637287, 0.43345377,\n",
       "        0.4385374 , 0.43943408, 0.43092865, 0.41575903, 0.42693686,\n",
       "        0.42078057, 0.4663294 , 0.40922865, 0.49892443, 0.3989337 ,\n",
       "        0.411878  , 0.4649778 , 0.43414208, 0.4137428 , 0.48975417,\n",
       "        0.4945829 , 0.40000975, 0.38872093, 0.45666113, 0.48274243,\n",
       "        0.4366221 , 0.4366221 , 0.4338407 , 0.46808898, 0.46823716,\n",
       "        0.43319237, 0.45413005, 0.4683492 , 0.41905022, 0.4871336 ,\n",
       "        0.42754486, 0.47478715, 0.4225916 , 0.4955297 , 0.39108995,\n",
       "        0.4680821 , 0.47549763, 0.46311387, 0.4223522 , 0.45357907,\n",
       "        0.41282848, 0.4774736 , 0.41278857, 0.44122964, 0.42254588,\n",
       "        0.47475073, 0.44197252, 0.45678687, 0.46340856, 0.4690892 ,\n",
       "        0.38732326, 0.3956504 , 0.44942185, 0.47272018, 0.45950457,\n",
       "        0.45944905, 0.48173988, 0.4441849 , 0.45612475, 0.42928952,\n",
       "        0.47999215, 0.46445537, 0.4859818 , 0.40877214, 0.4811168 ,\n",
       "        0.4797563 , 0.45382693, 0.45021355, 0.45906824, 0.38342622,\n",
       "        0.51035655, 0.48430791, 0.48568797, 0.4510436 , 0.49470747]],\n",
       "      dtype=float32)</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>t</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float32</div><div class='xr-var-preview xr-preview'>0.3074 0.3009 ... 0.3148 0.2974</div><input id='attrs-2e3e29c0-eedc-462f-8cd6-0674cb7cb51d' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-2e3e29c0-eedc-462f-8cd6-0674cb7cb51d' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-1dc94c4f-69d6-45ad-9480-437959442da8' class='xr-var-data-in' type='checkbox'><label for='data-1dc94c4f-69d6-45ad-9480-437959442da8' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.30737263, 0.30093196, 0.3084081 , 0.30146787, 0.30695248,\n",
       "        0.3091646 , 0.29918382, 0.3118719 , 0.30784422, 0.3068568 ,\n",
       "        0.3020741 , 0.30870333, 0.303456  , 0.3059026 , 0.30456638,\n",
       "        0.2999863 , 0.30318218, 0.30193773, 0.30662018, 0.30831078,\n",
       "        0.3033276 , 0.30293354, 0.30009118, 0.3083709 , 0.30206457,\n",
       "        0.3090841 , 0.30532503, 0.3051528 , 0.31389773, 0.29595286,\n",
       "        0.28398135, 0.2917807 , 0.29754576, 0.29689535, 0.30803528,\n",
       "        0.30520225, 0.30006847, 0.2993531 , 0.31028974, 0.3146022 ,\n",
       "        0.29532248, 0.31457052, 0.3114036 , 0.30938512, 0.30309147,\n",
       "        0.29534525, 0.30034715, 0.30678076, 0.30003032, 0.29394442,\n",
       "        0.3177059 , 0.29961377, 0.30048102, 0.31364104, 0.30861804,\n",
       "        0.30272946, 0.31413525, 0.29985458, 0.3039543 , 0.29880044,\n",
       "        0.30358857, 0.31341347, 0.3031626 , 0.3031626 , 0.30042604,\n",
       "        0.31699723, 0.30682147, 0.29895023, 0.30813107, 0.30391175,\n",
       "        0.30390027, 0.30651656, 0.3030581 , 0.31359392, 0.314182  ,\n",
       "        0.31131053, 0.31299418, 0.3117867 , 0.29921642, 0.29999846,\n",
       "        0.30702904, 0.30702904, 0.30043897, 0.30101788, 0.30518186,\n",
       "        0.30586335, 0.3017428 , 0.3004269 , 0.31018853, 0.30253905,\n",
       "        0.31260026, 0.29320493, 0.29726723, 0.29013437, 0.30612028,\n",
       "        0.30953848, 0.2987227 , 0.3075068 , 0.30406833, 0.30730686,\n",
       "...\n",
       "        0.29300973, 0.29858908, 0.29964802, 0.30467835, 0.30158833,\n",
       "        0.30549142, 0.30110997, 0.2992751 , 0.3065021 , 0.31931835,\n",
       "        0.30295405, 0.29834056, 0.29790005, 0.3066272 , 0.31092656,\n",
       "        0.30923328, 0.29882735, 0.293243  , 0.30026105, 0.3008416 ,\n",
       "        0.30504876, 0.3104526 , 0.30239373, 0.29423276, 0.2946543 ,\n",
       "        0.29205802, 0.2922547 , 0.31242988, 0.30049163, 0.30760592,\n",
       "        0.30801427, 0.3005932 , 0.30748472, 0.31191984, 0.29941717,\n",
       "        0.3029299 , 0.30860466, 0.3089271 , 0.29970363, 0.30067006,\n",
       "        0.30232865, 0.30232865, 0.3086319 , 0.2979667 , 0.3076866 ,\n",
       "        0.30475593, 0.29513964, 0.30392626, 0.30575776, 0.2990379 ,\n",
       "        0.31177455, 0.29968715, 0.30966866, 0.3015015 , 0.30876857,\n",
       "        0.30267444, 0.2993688 , 0.29483846, 0.3098791 , 0.30358252,\n",
       "        0.31467235, 0.2948168 , 0.30216792, 0.30131352, 0.3017997 ,\n",
       "        0.3037734 , 0.30480218, 0.3059065 , 0.30255044, 0.3034546 ,\n",
       "        0.30554518, 0.30720624, 0.30974236, 0.3064333 , 0.2978271 ,\n",
       "        0.2990801 , 0.3081027 , 0.29879916, 0.30681786, 0.30354407,\n",
       "        0.30678424, 0.29064187, 0.30324373, 0.30800706, 0.3066219 ,\n",
       "        0.3096639 , 0.29776916, 0.30465335, 0.30344725, 0.29410613,\n",
       "        0.31730413, 0.30699697, 0.307814  , 0.3148119 , 0.29736653]],\n",
       "      dtype=float32)</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>a</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float32</div><div class='xr-var-preview xr-preview'>1.246 1.253 1.286 ... 1.244 1.249</div><input id='attrs-87a515e8-d055-44e4-a7e1-da1e84877140' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-87a515e8-d055-44e4-a7e1-da1e84877140' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-3c5ad6dc-9c3c-49bd-afb9-b0cf0a929d0e' class='xr-var-data-in' type='checkbox'><label for='data-3c5ad6dc-9c3c-49bd-afb9-b0cf0a929d0e' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[1.2459027, 1.2528622, 1.2855899, 1.250947 , 1.2736264, 1.2804198,\n",
       "        1.2687066, 1.2578871, 1.2620679, 1.3015802, 1.2809064, 1.2899239,\n",
       "        1.2930617, 1.2760483, 1.3055099, 1.2410069, 1.2716178, 1.2628506,\n",
       "        1.2657764, 1.2367077, 1.3146054, 1.2679124, 1.2486309, 1.315798 ,\n",
       "        1.2565315, 1.2959132, 1.2777088, 1.2473727, 1.2471093, 1.3045945,\n",
       "        1.3717216, 1.3437476, 1.316791 , 1.3409469, 1.2775205, 1.2532446,\n",
       "        1.2670637, 1.2768476, 1.3689581, 1.3296181, 1.2825062, 1.2513225,\n",
       "        1.272394 , 1.2265012, 1.3242984, 1.3028877, 1.3511587, 1.2981392,\n",
       "        1.3215777, 1.339752 , 1.2992604, 1.2819748, 1.2867182, 1.3227768,\n",
       "        1.3332032, 1.243431 , 1.2672701, 1.3018801, 1.2881643, 1.2463281,\n",
       "        1.2800287, 1.2780293, 1.2924515, 1.2924515, 1.3443797, 1.2152562,\n",
       "        1.2875563, 1.2352194, 1.2862786, 1.2480134, 1.2459488, 1.2737472,\n",
       "        1.3021654, 1.2883388, 1.2637098, 1.214428 , 1.2606757, 1.2635279,\n",
       "        1.2797111, 1.2983618, 1.2598368, 1.2598368, 1.3211274, 1.2927781,\n",
       "        1.2349386, 1.2572556, 1.2506357, 1.2658932, 1.2922059, 1.2892894,\n",
       "        1.228254 , 1.2820951, 1.3149419, 1.298358 , 1.2722338, 1.3262312,\n",
       "        1.3112983, 1.2872958, 1.3081764, 1.3279549, 1.2427505, 1.309527 ,\n",
       "        1.2803055, 1.3113891, 1.272098 , 1.3001388, 1.2508657, 1.2699239,\n",
       "        1.2063452, 1.2037461, 1.2358239, 1.2797704, 1.2874552, 1.2827586,\n",
       "        1.2711511, 1.3159993, 1.2837677, 1.2462598, 1.3260257, 1.3028008,\n",
       "...\n",
       "        1.256544 , 1.2827086, 1.257236 , 1.2265772, 1.2037432, 1.3462031,\n",
       "        1.2257713, 1.2974963, 1.2580231, 1.2715585, 1.3080752, 1.3365114,\n",
       "        1.2302171, 1.2278885, 1.2278885, 1.2884241, 1.2646679, 1.3299048,\n",
       "        1.3327706, 1.2710243, 1.3078641, 1.2635862, 1.277014 , 1.2970521,\n",
       "        1.3000137, 1.2574576, 1.2881385, 1.3103478, 1.310604 , 1.2407393,\n",
       "        1.2363851, 1.3045386, 1.2452319, 1.2655917, 1.2688092, 1.2653241,\n",
       "        1.292524 , 1.2691543, 1.239563 , 1.2650944, 1.276172 , 1.2581555,\n",
       "        1.2669617, 1.3260275, 1.3050154, 1.3013009, 1.3109733, 1.3737149,\n",
       "        1.2133245, 1.315023 , 1.2448705, 1.2142311, 1.2747545, 1.2800554,\n",
       "        1.2936509, 1.270954 , 1.2808362, 1.2521682, 1.2353135, 1.35584  ,\n",
       "        1.2953225, 1.3187611, 1.3187611, 1.2424151, 1.2875434, 1.2409916,\n",
       "        1.2957673, 1.2727087, 1.3224319, 1.2354276, 1.2898556, 1.293252 ,\n",
       "        1.2803808, 1.2352489, 1.291453 , 1.2667351, 1.2626499, 1.3124354,\n",
       "        1.2651694, 1.3201593, 1.2554522, 1.2848926, 1.2598871, 1.3059498,\n",
       "        1.3152473, 1.2682987, 1.2616873, 1.2685087, 1.2837293, 1.2418935,\n",
       "        1.2493204, 1.280884 , 1.2958363, 1.2680354, 1.2363026, 1.2973623,\n",
       "        1.2993776, 1.3289416, 1.2781405, 1.3262049, 1.2287796, 1.293128 ,\n",
       "        1.3193277, 1.274465 , 1.2702065, 1.3332925, 1.3121424, 1.1998748,\n",
       "        1.2029315, 1.3317065, 1.3386822, 1.2262433, 1.2535826, 1.2464521,\n",
       "        1.2443454, 1.2493395]], dtype=float32)</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>v</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float32</div><div class='xr-var-preview xr-preview'>0.3329 0.3736 ... 0.4092 0.3667</div><input id='attrs-039888c2-12f6-4d63-a67b-22c65b34be7c' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-039888c2-12f6-4d63-a67b-22c65b34be7c' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-24ca1491-1c2d-4b16-b0af-5f29c59a68b8' class='xr-var-data-in' type='checkbox'><label for='data-24ca1491-1c2d-4b16-b0af-5f29c59a68b8' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.33285356, 0.37358427, 0.43430567, 0.34860754, 0.3762138 ,\n",
       "        0.3764081 , 0.3819971 , 0.3125658 , 0.34227753, 0.40833235,\n",
       "        0.33592033, 0.38616037, 0.44755125, 0.3946743 , 0.46904612,\n",
       "        0.32593298, 0.30748796, 0.402606  , 0.40332198, 0.4044881 ,\n",
       "        0.37006044, 0.34058976, 0.47196364, 0.3705516 , 0.39334965,\n",
       "        0.43107653, 0.315619  , 0.34416747, 0.37636995, 0.44288993,\n",
       "        0.42992353, 0.4549234 , 0.44793105, 0.37113953, 0.35303092,\n",
       "        0.38699102, 0.33790183, 0.42283845, 0.4475627 , 0.46044445,\n",
       "        0.37644148, 0.3572669 , 0.35901165, 0.3979075 , 0.4329052 ,\n",
       "        0.4192214 , 0.45369148, 0.42835522, 0.46734476, 0.45938873,\n",
       "        0.34250927, 0.44200134, 0.4299841 , 0.4456494 , 0.48070264,\n",
       "        0.33857346, 0.3663411 , 0.3665161 , 0.40260935, 0.32552862,\n",
       "        0.35689902, 0.34993672, 0.393744  , 0.393744  , 0.35881567,\n",
       "        0.26764512, 0.41193247, 0.30604458, 0.48057652, 0.3511176 ,\n",
       "        0.36841035, 0.4436295 , 0.38033676, 0.41709518, 0.37160683,\n",
       "        0.3487172 , 0.32357502, 0.3247769 , 0.4336996 , 0.40313292,\n",
       "        0.3893931 , 0.3893931 , 0.42867517, 0.4392171 , 0.37201953,\n",
       "        0.33844423, 0.41021585, 0.2932155 , 0.4203651 , 0.47055626,\n",
       "        0.32935166, 0.38881016, 0.41675615, 0.41484165, 0.43766832,\n",
       "        0.41378617, 0.41991472, 0.39268064, 0.351933  , 0.38750887,\n",
       "...\n",
       "        0.38404608, 0.33950806, 0.33850384, 0.39999866, 0.44818592,\n",
       "        0.38812923, 0.3818829 , 0.41476178, 0.29965544, 0.37407064,\n",
       "        0.35495496, 0.3439827 , 0.33925223, 0.38588667, 0.44738674,\n",
       "        0.4620142 , 0.36589956, 0.37878323, 0.377707  , 0.3529315 ,\n",
       "        0.45715356, 0.3968935 , 0.44173908, 0.4623909 , 0.46347666,\n",
       "        0.4313407 , 0.39593935, 0.40275335, 0.31843972, 0.4439199 ,\n",
       "        0.47216797, 0.36289263, 0.46282268, 0.4391365 , 0.3338251 ,\n",
       "        0.29173708, 0.47126985, 0.40762186, 0.39030457, 0.346622  ,\n",
       "        0.42770576, 0.42770576, 0.39899755, 0.35091734, 0.3440752 ,\n",
       "        0.42559528, 0.3620839 , 0.408782  , 0.4561429 , 0.35695744,\n",
       "        0.39797688, 0.35332823, 0.4125204 , 0.33260584, 0.49025202,\n",
       "        0.32782173, 0.42521524, 0.33179522, 0.4225502 , 0.43363166,\n",
       "        0.47817087, 0.37301636, 0.46000433, 0.4136796 , 0.4409461 ,\n",
       "        0.3461988 , 0.37007475, 0.4309759 , 0.3352933 , 0.34006405,\n",
       "        0.38962555, 0.45559025, 0.37462735, 0.29401517, 0.3755951 ,\n",
       "        0.38964486, 0.41810632, 0.36719656, 0.38397217, 0.38319921,\n",
       "        0.39509058, 0.37683678, 0.38905692, 0.421391  , 0.41299057,\n",
       "        0.4323635 , 0.4162984 , 0.34719276, 0.41606808, 0.5440178 ,\n",
       "        0.28669667, 0.27096653, 0.31075907, 0.40919876, 0.36665344]],\n",
       "      dtype=float32)</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-da30446b-2891-404d-a24e-e7caf9c85a36' class='xr-section-summary-in' type='checkbox'  ><label for='section-da30446b-2891-404d-a24e-e7caf9c85a36' class='xr-section-summary' >Indexes: <span>(2)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-index-name'><div>chain</div></div><div class='xr-index-preview'>PandasIndex</div><div></div><input id='index-387cf240-8214-4c13-bb56-459fdafbfbcb' class='xr-index-data-in' type='checkbox'/><label for='index-387cf240-8214-4c13-bb56-459fdafbfbcb' title='Show/Hide index repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-index-data'><pre>PandasIndex(Index([0, 1], dtype=&#x27;int64&#x27;, name=&#x27;chain&#x27;))</pre></div></li><li class='xr-var-item'><div class='xr-index-name'><div>draw</div></div><div class='xr-index-preview'>PandasIndex</div><div></div><input id='index-f9f2892c-4adf-4fa5-a235-722e1527bccc' class='xr-index-data-in' type='checkbox'/><label for='index-f9f2892c-4adf-4fa5-a235-722e1527bccc' title='Show/Hide index repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-index-data'><pre>PandasIndex(Index([  0,   1,   2,   3,   4,   5,   6,   7,   8,   9,\n",
       "       ...\n",
       "       490, 491, 492, 493, 494, 495, 496, 497, 498, 499],\n",
       "      dtype=&#x27;int64&#x27;, name=&#x27;draw&#x27;, length=500))</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-ec5ef896-5fe9-40c7-a554-272911fc6a87' class='xr-section-summary-in' type='checkbox'  checked><label for='section-ec5ef896-5fe9-40c7-a554-272911fc6a87' class='xr-section-summary' >Attributes: <span>(8)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><dl class='xr-attrs'><dt><span>created_at :</span></dt><dd>2023-06-22T18:12:24.938346</dd><dt><span>arviz_version :</span></dt><dd>0.14.0</dd><dt><span>inference_library :</span></dt><dd>pymc</dd><dt><span>inference_library_version :</span></dt><dd>5.5.0</dd><dt><span>sampling_time :</span></dt><dd>35.316415786743164</dd><dt><span>tuning_steps :</span></dt><dd>500</dd><dt><span>modeling_interface :</span></dt><dd>bambi</dd><dt><span>modeling_interface_version :</span></dt><dd>0.11.0</dd></dl></div></li></ul></div></div><br></div>\n",
       "                      </ul>\n",
       "                  </div>\n",
       "            </li>\n",
       "            \n",
       "            <li class = \"xr-section-item\">\n",
       "                  <input id=\"idata_sample_stats9dd1645b-f1de-42c8-9e2c-9e5f430accd9\" class=\"xr-section-summary-in\" type=\"checkbox\">\n",
       "                  <label for=\"idata_sample_stats9dd1645b-f1de-42c8-9e2c-9e5f430accd9\" class = \"xr-section-summary\">sample_stats</label>\n",
       "                  <div class=\"xr-section-inline-details\"></div>\n",
       "                  <div class=\"xr-section-details\">\n",
       "                      <ul id=\"xr-dataset-coord-list\" class=\"xr-var-list\">\n",
       "                          <div style=\"padding-left:2rem;\"><div><svg style=\"position: absolute; width: 0; height: 0; overflow: hidden\">\n",
       "<defs>\n",
       "<symbol id=\"icon-database\" viewBox=\"0 0 32 32\">\n",
       "<path d=\"M16 0c-8.837 0-16 2.239-16 5v4c0 2.761 7.163 5 16 5s16-2.239 16-5v-4c0-2.761-7.163-5-16-5z\"></path>\n",
       "<path d=\"M16 17c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n",
       "<path d=\"M16 26c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n",
       "</symbol>\n",
       "<symbol id=\"icon-file-text2\" viewBox=\"0 0 32 32\">\n",
       "<path d=\"M28.681 7.159c-0.694-0.947-1.662-2.053-2.724-3.116s-2.169-2.030-3.116-2.724c-1.612-1.182-2.393-1.319-2.841-1.319h-15.5c-1.378 0-2.5 1.121-2.5 2.5v27c0 1.378 1.122 2.5 2.5 2.5h23c1.378 0 2.5-1.122 2.5-2.5v-19.5c0-0.448-0.137-1.23-1.319-2.841zM24.543 5.457c0.959 0.959 1.712 1.825 2.268 2.543h-4.811v-4.811c0.718 0.556 1.584 1.309 2.543 2.268zM28 29.5c0 0.271-0.229 0.5-0.5 0.5h-23c-0.271 0-0.5-0.229-0.5-0.5v-27c0-0.271 0.229-0.5 0.5-0.5 0 0 15.499-0 15.5 0v7c0 0.552 0.448 1 1 1h7v19.5z\"></path>\n",
       "<path d=\"M23 26h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "<path d=\"M23 22h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "<path d=\"M23 18h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "</symbol>\n",
       "</defs>\n",
       "</svg>\n",
       "<style>/* CSS stylesheet for displaying xarray objects in jupyterlab.\n",
       " *\n",
       " */\n",
       "\n",
       ":root {\n",
       "  --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));\n",
       "  --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));\n",
       "  --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));\n",
       "  --xr-border-color: var(--jp-border-color2, #e0e0e0);\n",
       "  --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);\n",
       "  --xr-background-color: var(--jp-layout-color0, white);\n",
       "  --xr-background-color-row-even: var(--jp-layout-color1, white);\n",
       "  --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);\n",
       "}\n",
       "\n",
       "html[theme=dark],\n",
       "body[data-theme=dark],\n",
       "body.vscode-dark {\n",
       "  --xr-font-color0: rgba(255, 255, 255, 1);\n",
       "  --xr-font-color2: rgba(255, 255, 255, 0.54);\n",
       "  --xr-font-color3: rgba(255, 255, 255, 0.38);\n",
       "  --xr-border-color: #1F1F1F;\n",
       "  --xr-disabled-color: #515151;\n",
       "  --xr-background-color: #111111;\n",
       "  --xr-background-color-row-even: #111111;\n",
       "  --xr-background-color-row-odd: #313131;\n",
       "}\n",
       "\n",
       ".xr-wrap {\n",
       "  display: block !important;\n",
       "  min-width: 300px;\n",
       "  max-width: 700px;\n",
       "}\n",
       "\n",
       ".xr-text-repr-fallback {\n",
       "  /* fallback to plain text repr when CSS is not injected (untrusted notebook) */\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-header {\n",
       "  padding-top: 6px;\n",
       "  padding-bottom: 6px;\n",
       "  margin-bottom: 4px;\n",
       "  border-bottom: solid 1px var(--xr-border-color);\n",
       "}\n",
       "\n",
       ".xr-header > div,\n",
       ".xr-header > ul {\n",
       "  display: inline;\n",
       "  margin-top: 0;\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-obj-type,\n",
       ".xr-array-name {\n",
       "  margin-left: 2px;\n",
       "  margin-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-obj-type {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-sections {\n",
       "  padding-left: 0 !important;\n",
       "  display: grid;\n",
       "  grid-template-columns: 150px auto auto 1fr 20px 20px;\n",
       "}\n",
       "\n",
       ".xr-section-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-section-item input {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-item input + label {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label {\n",
       "  cursor: pointer;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label:hover {\n",
       "  color: var(--xr-font-color0);\n",
       "}\n",
       "\n",
       ".xr-section-summary {\n",
       "  grid-column: 1;\n",
       "  color: var(--xr-font-color2);\n",
       "  font-weight: 500;\n",
       "}\n",
       "\n",
       ".xr-section-summary > span {\n",
       "  display: inline-block;\n",
       "  padding-left: 0.5em;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in + label:before {\n",
       "  display: inline-block;\n",
       "  content: '►';\n",
       "  font-size: 11px;\n",
       "  width: 15px;\n",
       "  text-align: center;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label:before {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label:before {\n",
       "  content: '▼';\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label > span {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-summary,\n",
       ".xr-section-inline-details {\n",
       "  padding-top: 4px;\n",
       "  padding-bottom: 4px;\n",
       "}\n",
       "\n",
       ".xr-section-inline-details {\n",
       "  grid-column: 2 / -1;\n",
       "}\n",
       "\n",
       ".xr-section-details {\n",
       "  display: none;\n",
       "  grid-column: 1 / -1;\n",
       "  margin-bottom: 5px;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked ~ .xr-section-details {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-array-wrap {\n",
       "  grid-column: 1 / -1;\n",
       "  display: grid;\n",
       "  grid-template-columns: 20px auto;\n",
       "}\n",
       "\n",
       ".xr-array-wrap > label {\n",
       "  grid-column: 1;\n",
       "  vertical-align: top;\n",
       "}\n",
       "\n",
       ".xr-preview {\n",
       "  color: var(--xr-font-color3);\n",
       "}\n",
       "\n",
       ".xr-array-preview,\n",
       ".xr-array-data {\n",
       "  padding: 0 5px !important;\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-array-data,\n",
       ".xr-array-in:checked ~ .xr-array-preview {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-array-in:checked ~ .xr-array-data,\n",
       ".xr-array-preview {\n",
       "  display: inline-block;\n",
       "}\n",
       "\n",
       ".xr-dim-list {\n",
       "  display: inline-block !important;\n",
       "  list-style: none;\n",
       "  padding: 0 !important;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list li {\n",
       "  display: inline-block;\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list:before {\n",
       "  content: '(';\n",
       "}\n",
       "\n",
       ".xr-dim-list:after {\n",
       "  content: ')';\n",
       "}\n",
       "\n",
       ".xr-dim-list li:not(:last-child):after {\n",
       "  content: ',';\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-has-index {\n",
       "  font-weight: bold;\n",
       "}\n",
       "\n",
       ".xr-var-list,\n",
       ".xr-var-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-var-item > div,\n",
       ".xr-var-item label,\n",
       ".xr-var-item > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-even);\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-var-item > .xr-var-name:hover span {\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-var-list > li:nth-child(odd) > div,\n",
       ".xr-var-list > li:nth-child(odd) > label,\n",
       ".xr-var-list > li:nth-child(odd) > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-odd);\n",
       "}\n",
       "\n",
       ".xr-var-name {\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-var-dims {\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-var-dtype {\n",
       "  grid-column: 3;\n",
       "  text-align: right;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-preview {\n",
       "  grid-column: 4;\n",
       "}\n",
       "\n",
       ".xr-index-preview {\n",
       "  grid-column: 2 / 5;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-name,\n",
       ".xr-var-dims,\n",
       ".xr-var-dtype,\n",
       ".xr-preview,\n",
       ".xr-attrs dt {\n",
       "  white-space: nowrap;\n",
       "  overflow: hidden;\n",
       "  text-overflow: ellipsis;\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-var-name:hover,\n",
       ".xr-var-dims:hover,\n",
       ".xr-var-dtype:hover,\n",
       ".xr-attrs dt:hover {\n",
       "  overflow: visible;\n",
       "  width: auto;\n",
       "  z-index: 1;\n",
       "}\n",
       "\n",
       ".xr-var-attrs,\n",
       ".xr-var-data,\n",
       ".xr-index-data {\n",
       "  display: none;\n",
       "  background-color: var(--xr-background-color) !important;\n",
       "  padding-bottom: 5px !important;\n",
       "}\n",
       "\n",
       ".xr-var-attrs-in:checked ~ .xr-var-attrs,\n",
       ".xr-var-data-in:checked ~ .xr-var-data,\n",
       ".xr-index-data-in:checked ~ .xr-index-data {\n",
       "  display: block;\n",
       "}\n",
       "\n",
       ".xr-var-data > table {\n",
       "  float: right;\n",
       "}\n",
       "\n",
       ".xr-var-name span,\n",
       ".xr-var-data,\n",
       ".xr-index-name div,\n",
       ".xr-index-data,\n",
       ".xr-attrs {\n",
       "  padding-left: 25px !important;\n",
       "}\n",
       "\n",
       ".xr-attrs,\n",
       ".xr-var-attrs,\n",
       ".xr-var-data,\n",
       ".xr-index-data {\n",
       "  grid-column: 1 / -1;\n",
       "}\n",
       "\n",
       "dl.xr-attrs {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  display: grid;\n",
       "  grid-template-columns: 125px auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt,\n",
       ".xr-attrs dd {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  float: left;\n",
       "  padding-right: 10px;\n",
       "  width: auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt {\n",
       "  font-weight: normal;\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-attrs dt:hover span {\n",
       "  display: inline-block;\n",
       "  background: var(--xr-background-color);\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-attrs dd {\n",
       "  grid-column: 2;\n",
       "  white-space: pre-wrap;\n",
       "  word-break: break-all;\n",
       "}\n",
       "\n",
       ".xr-icon-database,\n",
       ".xr-icon-file-text2,\n",
       ".xr-no-icon {\n",
       "  display: inline-block;\n",
       "  vertical-align: middle;\n",
       "  width: 1em;\n",
       "  height: 1.5em !important;\n",
       "  stroke-width: 0;\n",
       "  stroke: currentColor;\n",
       "  fill: currentColor;\n",
       "}\n",
       "</style><pre class='xr-text-repr-fallback'>&lt;xarray.Dataset&gt;\n",
       "Dimensions:                (chain: 2, draw: 500)\n",
       "Coordinates:\n",
       "  * chain                  (chain) int64 0 1\n",
       "  * draw                   (draw) int64 0 1 2 3 4 5 ... 494 495 496 497 498 499\n",
       "Data variables: (12/17)\n",
       "    n_steps                (chain, draw) float64 7.0 3.0 7.0 7.0 ... 7.0 7.0 7.0\n",
       "    largest_eigval         (chain, draw) float64 nan nan nan nan ... nan nan nan\n",
       "    perf_counter_start     (chain, draw) float64 1.484e+05 ... 1.484e+05\n",
       "    reached_max_treedepth  (chain, draw) bool False False False ... False False\n",
       "    step_size              (chain, draw) float64 0.6616 0.6616 ... 0.6886 0.6886\n",
       "    step_size_bar          (chain, draw) float64 0.7015 0.7015 ... 0.5939 0.5939\n",
       "    ...                     ...\n",
       "    perf_counter_diff      (chain, draw) float64 0.01556 0.008141 ... 0.01593\n",
       "    diverging              (chain, draw) bool False False False ... False False\n",
       "    energy                 (chain, draw) float64 1.867e+03 ... 1.873e+03\n",
       "    smallest_eigval        (chain, draw) float64 nan nan nan nan ... nan nan nan\n",
       "    acceptance_rate        (chain, draw) float64 0.8446 0.8703 ... 0.6986 0.3913\n",
       "    process_time_diff      (chain, draw) float64 0.03068 0.01581 ... 0.03008\n",
       "Attributes:\n",
       "    created_at:                  2023-06-22T18:12:24.942485\n",
       "    arviz_version:               0.14.0\n",
       "    inference_library:           pymc\n",
       "    inference_library_version:   5.5.0\n",
       "    sampling_time:               35.316415786743164\n",
       "    tuning_steps:                500\n",
       "    modeling_interface:          bambi\n",
       "    modeling_interface_version:  0.11.0</pre><div class='xr-wrap' style='display:none'><div class='xr-header'><div class='xr-obj-type'>xarray.Dataset</div></div><ul class='xr-sections'><li class='xr-section-item'><input id='section-3f3d2f45-2bad-4452-b0a2-545bd1411e63' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-3f3d2f45-2bad-4452-b0a2-545bd1411e63' class='xr-section-summary'  title='Expand/collapse section'>Dimensions:</label><div class='xr-section-inline-details'><ul class='xr-dim-list'><li><span class='xr-has-index'>chain</span>: 2</li><li><span class='xr-has-index'>draw</span>: 500</li></ul></div><div class='xr-section-details'></div></li><li class='xr-section-item'><input id='section-9f2a412e-0a74-4162-85c8-6e62804ab912' class='xr-section-summary-in' type='checkbox'  checked><label for='section-9f2a412e-0a74-4162-85c8-6e62804ab912' class='xr-section-summary' >Coordinates: <span>(2)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span class='xr-has-index'>chain</span></div><div class='xr-var-dims'>(chain)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>0 1</div><input id='attrs-ff180b1b-e637-44c4-b0c6-f15cbe84362b' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-ff180b1b-e637-44c4-b0c6-f15cbe84362b' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-0391b5e6-eba3-4873-97fd-9241732c6fe8' class='xr-var-data-in' type='checkbox'><label for='data-0391b5e6-eba3-4873-97fd-9241732c6fe8' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([0, 1])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span class='xr-has-index'>draw</span></div><div class='xr-var-dims'>(draw)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>0 1 2 3 4 5 ... 495 496 497 498 499</div><input id='attrs-badb3b04-0d6e-45b3-924a-7ee29968b72c' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-badb3b04-0d6e-45b3-924a-7ee29968b72c' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-69ea2f38-6646-4dfa-8f64-210acd8481f7' class='xr-var-data-in' type='checkbox'><label for='data-69ea2f38-6646-4dfa-8f64-210acd8481f7' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([  0,   1,   2, ..., 497, 498, 499])</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-d1ecce6e-c0ca-4110-98af-edd521cecc37' class='xr-section-summary-in' type='checkbox'  ><label for='section-d1ecce6e-c0ca-4110-98af-edd521cecc37' class='xr-section-summary' >Data variables: <span>(17)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span>n_steps</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>7.0 3.0 7.0 7.0 ... 3.0 7.0 7.0 7.0</div><input id='attrs-8ffd0d23-63b9-4633-82bc-01bdb46d6000' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-8ffd0d23-63b9-4633-82bc-01bdb46d6000' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-f8094f1e-d911-4831-8b00-9da04fdd3b76' class='xr-var-data-in' type='checkbox'><label for='data-f8094f1e-d911-4831-8b00-9da04fdd3b76' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[7., 3., 7., 7., 3., 7., 7., 7., 3., 7., 7., 7., 3., 7., 7., 7.,\n",
       "        3., 7., 3., 7., 7., 7., 3., 3., 7., 7., 7., 3., 7., 7., 3., 3.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 3., 7., 3., 3., 3.,\n",
       "        7., 3., 7., 7., 1., 3., 7., 7., 7., 7., 3., 3., 7., 3., 7., 3.,\n",
       "        3., 7., 7., 7., 7., 7., 7., 7., 3., 7., 3., 7., 7., 7., 7., 3.,\n",
       "        7., 3., 7., 3., 7., 3., 3., 3., 7., 3., 7., 7., 7., 7., 7., 3.,\n",
       "        7., 7., 3., 3., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7.,\n",
       "        7., 3., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 3., 3.,\n",
       "        7., 7., 3., 7., 7., 7., 7., 3., 7., 7., 3., 7., 7., 7., 7., 3.,\n",
       "        7., 7., 7., 3., 7., 7., 3., 7., 7., 3., 3., 7., 7., 7., 7., 7.,\n",
       "        7., 3., 3., 3., 7., 3., 7., 3., 7., 3., 7., 7., 7., 7., 7., 7.,\n",
       "        3., 7., 7., 7., 7., 3., 7., 7., 7., 7., 7., 7., 3., 7., 7., 3.,\n",
       "        7., 7., 7., 3., 7., 7., 7., 7., 7., 3., 7., 7., 7., 7., 3., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 3., 3., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 3., 7., 7., 7., 3.,\n",
       "        3., 7., 3., 7., 7., 7., 3., 3., 7., 7., 7., 3., 7., 7., 7., 7.,\n",
       "        7., 7., 7., 3., 3., 7., 7., 7., 7., 7., 7., 5., 7., 7., 7., 3.,\n",
       "        7., 7., 7., 3., 3., 7., 3., 3., 7., 7., 7., 1., 3., 3., 7., 7.,\n",
       "        7., 7., 3., 7., 7., 7., 7., 7., 3., 3., 7., 7., 7., 3., 3., 3.,\n",
       "...\n",
       "        3., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 3.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 3., 7., 7., 7., 7., 7., 7., 7., 3., 7., 3., 3.,\n",
       "        3., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 7., 3.,\n",
       "        3., 7., 7., 7., 7., 7., 3., 7., 3., 7., 3., 7., 7., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 7., 7., 7., 7., 7.,\n",
       "        7., 3., 7., 7., 7., 7., 3., 7., 3., 7., 7., 7., 7., 7., 7., 3.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 7.,\n",
       "        7., 3., 7., 7., 7., 3., 7., 3., 7., 7., 7., 3., 3., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 3.,\n",
       "        7., 7., 7., 7., 3., 7., 3., 7., 7., 7., 7., 7., 7., 7., 3., 7.,\n",
       "        7., 7., 7., 3., 3., 7., 7., 7., 3., 7., 3., 3., 7., 3., 3., 3.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 3., 7., 7., 7., 7.,\n",
       "        7., 7., 3., 7., 7., 3., 7., 7., 3., 7., 3., 7., 7., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 3., 7., 7., 7., 3., 7., 7., 7., 7., 3., 3., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 3., 7., 3., 7., 3., 7., 7., 7., 7., 7.,\n",
       "        7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7.,\n",
       "        7., 3., 7., 7., 7., 3., 7., 7., 7., 7., 3., 7., 7., 7., 7., 7.,\n",
       "        3., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7., 7.,\n",
       "        3., 7., 7., 7.]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>largest_eigval</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>nan nan nan nan ... nan nan nan nan</div><input id='attrs-76a2eaa9-9956-4b90-ad32-194792d544d7' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-76a2eaa9-9956-4b90-ad32-194792d544d7' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-6ab1a0ed-5902-42fb-b270-40fd972db467' class='xr-var-data-in' type='checkbox'><label for='data-6ab1a0ed-5902-42fb-b270-40fd972db467' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "...\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>perf_counter_start</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>1.484e+05 1.484e+05 ... 1.484e+05</div><input id='attrs-aa7353fa-28ab-43fd-9b5b-1a063e290b1b' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-aa7353fa-28ab-43fd-9b5b-1a063e290b1b' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-51e1e9fe-c4fa-49e5-859f-2de952e196eb' class='xr-var-data-in' type='checkbox'><label for='data-51e1e9fe-c4fa-49e5-859f-2de952e196eb' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[148370.23824617, 148370.25392321, 148370.26216571,\n",
       "        148370.27863688, 148370.29496775, 148370.30291704,\n",
       "        148370.31952642, 148370.33551908, 148370.35153208,\n",
       "        148370.35990504, 148370.37696625, 148370.39320704,\n",
       "        148370.40946613, 148370.41737571, 148370.43377683,\n",
       "        148370.44943171, 148370.46611921, 148370.47448392,\n",
       "        148370.48995458, 148370.49799267, 148370.51413875,\n",
       "        148370.52964113, 148370.54581413, 148370.55432133,\n",
       "        148370.56235129, 148370.57843025, 148370.59434887,\n",
       "        148370.61497917, 148370.62350663, 148370.63984021,\n",
       "        148370.65601867, 148370.66418979, 148370.67226921,\n",
       "        148370.68848946, 148370.70462154, 148370.72035867,\n",
       "        148370.73597421, 148370.75215338, 148370.76875217,\n",
       "        148370.78402617, 148370.79244238, 148370.80843533,\n",
       "        148370.82455775, 148370.84037858, 148370.84830912,\n",
       "        148370.86469283, 148370.87313733, 148370.88140908,\n",
       "        148370.88958471, 148370.90575504, 148370.9137915 ,\n",
       "        148370.93014654, 148370.946414  , 148370.95059225,\n",
       "        148370.95902242, 148370.97548108, 148370.99142579,\n",
       "        148371.00767913, 148371.02459654, 148371.03325688,\n",
       "...\n",
       "        148378.9797575 , 148378.99519846, 148379.00316017,\n",
       "        148379.01861987, 148379.03532737, 148379.05114379,\n",
       "        148379.06684229, 148379.08197733, 148379.09788392,\n",
       "        148379.11414308, 148379.12971562, 148379.14535392,\n",
       "        148379.16052696, 148379.17582958, 148379.19170025,\n",
       "        148379.20716283, 148379.22261542, 148379.23786421,\n",
       "        148379.25381483, 148379.26909988, 148379.28477721,\n",
       "        148379.29964663, 148379.31509929, 148379.33089554,\n",
       "        148379.34639258, 148379.35422887, 148379.36965813,\n",
       "        148379.38550621, 148379.4011115 , 148379.4090955 ,\n",
       "        148379.424652  , 148379.44017246, 148379.45528475,\n",
       "        148379.470358  , 148379.47826625, 148379.49397842,\n",
       "        148379.50948996, 148379.52478042, 148379.54041829,\n",
       "        148379.555854  , 148379.56382971, 148379.57960371,\n",
       "        148379.59510621, 148379.61035396, 148379.62586063,\n",
       "        148379.64067504, 148379.65613133, 148379.67110021,\n",
       "        148379.68713179, 148379.70304271, 148379.71884937,\n",
       "        148379.73453542, 148379.75014867, 148379.76496433,\n",
       "        148379.7801195 , 148379.79518679, 148379.80283637,\n",
       "        148379.81844821, 148379.83425412]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>reached_max_treedepth</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>bool</div><div class='xr-var-preview xr-preview'>False False False ... False False</div><input id='attrs-419141c0-4db5-4cbe-b803-960f16fbc2bc' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-419141c0-4db5-4cbe-b803-960f16fbc2bc' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-8819940d-d01d-4a7b-a95f-688d7b4079f9' class='xr-var-data-in' type='checkbox'><label for='data-8819940d-d01d-4a7b-a95f-688d7b4079f9' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "...\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>step_size</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.6616 0.6616 ... 0.6886 0.6886</div><input id='attrs-b999853b-0330-4495-a845-514c7fa7ce57' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-b999853b-0330-4495-a845-514c7fa7ce57' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-e8853546-9164-43ec-9a53-28aa36e15e2b' class='xr-var-data-in' type='checkbox'><label for='data-e8853546-9164-43ec-9a53-28aa36e15e2b' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "        0.66164449, 0.66164449, 0.66164449, 0.66164449, 0.66164449,\n",
       "...\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402,\n",
       "        0.68857402, 0.68857402, 0.68857402, 0.68857402, 0.68857402]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>step_size_bar</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.7015 0.7015 ... 0.5939 0.5939</div><input id='attrs-73f63b84-d065-48d3-8c91-7ebeb795f1e7' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-73f63b84-d065-48d3-8c91-7ebeb795f1e7' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-a0b05203-e997-4119-87c3-78a552567873' class='xr-var-data-in' type='checkbox'><label for='data-a0b05203-e997-4119-87c3-78a552567873' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "        0.70151793, 0.70151793, 0.70151793, 0.70151793, 0.70151793,\n",
       "...\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ,\n",
       "        0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 , 0.5938913 ]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>max_energy_error</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.2456 0.258 ... 0.7157 1.595</div><input id='attrs-06ac66b4-0b7e-4c41-a204-63c1ff73dd48' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-06ac66b4-0b7e-4c41-a204-63c1ff73dd48' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-be623bdb-68d1-484c-a073-6fb88e7eb9d5' class='xr-var-data-in' type='checkbox'><label for='data-be623bdb-68d1-484c-a073-6fb88e7eb9d5' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[ 2.45618641e-01,  2.58033089e-01,  1.06561646e-01,\n",
       "         1.64673924e-01,  3.30347449e-01,  1.73879504e-01,\n",
       "         3.74828935e-01, -4.39571619e-01, -3.54073942e-01,\n",
       "         5.90958476e-01, -3.75126600e-01, -4.60566521e-01,\n",
       "         6.45961380e+00,  1.03453267e+00,  4.03460979e-01,\n",
       "        -2.10318565e-01, -2.35695100e+00,  1.96818280e+00,\n",
       "        -3.38342786e-02, -2.74529576e-01, -1.66865993e+00,\n",
       "         8.49430799e-01, -4.51771259e-01, -1.06480360e+00,\n",
       "        -6.41608059e-01,  7.15437770e-01,  2.98294902e-01,\n",
       "         3.64206076e-01,  3.31632018e-01,  1.79381037e+00,\n",
       "         8.16988707e-01,  1.29253125e+00, -5.30653715e-01,\n",
       "         2.02202582e+00, -1.20626330e+00,  5.24350286e-01,\n",
       "        -3.18356991e-01,  1.23666644e-01, -3.29166174e-01,\n",
       "        -2.09434271e-01, -3.40318680e-01,  1.14394307e-01,\n",
       "         4.37180281e-01,  5.69258690e-01, -7.91580677e-02,\n",
       "         1.58238888e-01,  4.96041536e-01,  4.16967630e-01,\n",
       "         4.04501081e-01,  1.94438934e-01,  3.32537174e-01,\n",
       "         7.24316001e-01, -1.47991717e-01,  5.96048117e-01,\n",
       "        -6.40288591e-01,  1.44761086e-01, -2.34005809e-01,\n",
       "        -2.85876155e-01,  5.03491104e-01, -3.32464457e-01,\n",
       "...\n",
       "        -2.50544071e-01,  5.38447857e-01, -3.54901791e-01,\n",
       "         4.77323675e+00, -9.21057463e-02,  1.44683218e+00,\n",
       "         5.76945305e-01, -5.25576830e-01,  1.15809679e-01,\n",
       "         7.87444115e-02,  1.37336791e-01, -8.34319592e-02,\n",
       "         2.69174790e+00, -2.62360334e-01,  2.71673173e-01,\n",
       "        -2.71156579e-01,  3.57160926e-01,  2.83486784e-01,\n",
       "        -2.18018770e-01, -2.73572087e-01,  5.76207161e-01,\n",
       "        -3.45818758e-01, -1.80367589e-01,  3.39036345e-01,\n",
       "         2.51382136e+00,  9.46527898e-01, -2.79981852e-01,\n",
       "         1.74045563e-01,  2.99218297e-01,  1.64448887e-01,\n",
       "         3.36850882e-01,  4.39037263e-01, -2.25873947e-01,\n",
       "        -2.40705758e-02,  2.98824185e+00, -1.10505772e+00,\n",
       "        -2.41398692e-01,  2.93053150e-01,  4.77325678e-01,\n",
       "        -8.37546736e-02,  1.06197166e+00,  1.15921330e+00,\n",
       "         7.52879858e-01,  7.20850229e-01,  3.19597900e-01,\n",
       "         7.66227335e-01,  1.27523750e+00, -5.24915934e-01,\n",
       "         1.21006846e-01,  4.59831715e-01, -5.28943062e-01,\n",
       "        -1.04033184e+00,  1.35831892e-01,  2.47248888e-01,\n",
       "        -1.81982279e-01,  2.51667142e-01, -4.18622494e-01,\n",
       "         7.15735555e-01,  1.59498310e+00]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>lp</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>-1.865e+03 ... -1.869e+03</div><input id='attrs-b986059d-a358-43ae-99cf-3dd4c778d992' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-b986059d-a358-43ae-99cf-3dd4c778d992' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-46962c94-249e-4180-91a7-a0a67bb6a6a3' class='xr-var-data-in' type='checkbox'><label for='data-46962c94-249e-4180-91a7-a0a67bb6a6a3' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[-1865.30944824, -1864.33300781, -1864.39099121, -1864.71984863,\n",
       "        -1863.95959473, -1864.73156738, -1866.01867676, -1866.04394531,\n",
       "        -1864.59313965, -1865.56298828, -1865.55578613, -1865.65966797,\n",
       "        -1867.01977539, -1865.22729492, -1865.51196289, -1865.34155273,\n",
       "        -1868.22302246, -1868.3885498 , -1867.70068359, -1869.37744141,\n",
       "        -1865.05871582, -1867.42956543, -1867.71813965, -1866.89208984,\n",
       "        -1865.67089844, -1866.41442871, -1866.46630859, -1864.75683594,\n",
       "        -1867.22485352, -1866.04882812, -1870.33825684, -1868.81323242,\n",
       "        -1865.91516113, -1868.38720703, -1864.92871094, -1865.96411133,\n",
       "        -1865.48657227, -1866.2166748 , -1867.71130371, -1867.1517334 ,\n",
       "        -1865.01916504, -1865.72180176, -1865.41235352, -1865.3190918 ,\n",
       "        -1864.9909668 , -1865.61523438, -1865.93688965, -1864.98596191,\n",
       "        -1865.54345703, -1867.1652832 , -1867.55493164, -1865.11425781,\n",
       "        -1864.59436035, -1868.22644043, -1866.72814941, -1865.69384766,\n",
       "        -1865.55310059, -1865.16540527, -1865.31738281, -1865.39477539,\n",
       "        -1864.22497559, -1865.51416016, -1864.8560791 , -1864.8560791 ,\n",
       "        -1866.51257324, -1868.69909668, -1864.35400391, -1870.03796387,\n",
       "        -1865.58605957, -1864.31225586, -1864.30285645, -1864.64050293,\n",
       "        -1867.4675293 , -1865.49133301, -1864.98156738, -1866.01879883,\n",
       "        -1868.51550293, -1866.09790039, -1864.78259277, -1865.32397461,\n",
       "...\n",
       "        -1865.33544922, -1866.70715332, -1866.29821777, -1864.92297363,\n",
       "        -1865.38037109, -1866.89770508, -1865.00805664, -1865.48327637,\n",
       "        -1866.45605469, -1866.04797363, -1866.53234863, -1868.54602051,\n",
       "        -1867.09533691, -1866.82702637, -1866.19299316, -1868.45471191,\n",
       "        -1865.64953613, -1867.22424316, -1867.8737793 , -1865.43212891,\n",
       "        -1867.55810547, -1867.52783203, -1867.86181641, -1868.05249023,\n",
       "        -1865.35070801, -1865.43884277, -1865.43884277, -1865.92602539,\n",
       "        -1864.84655762, -1864.5255127 , -1864.3248291 , -1865.34191895,\n",
       "        -1865.12731934, -1865.98217773, -1865.46057129, -1865.28979492,\n",
       "        -1864.70373535, -1865.26281738, -1866.14672852, -1867.41394043,\n",
       "        -1865.18078613, -1866.34301758, -1866.47045898, -1865.9543457 ,\n",
       "        -1865.27929688, -1866.86938477, -1867.62927246, -1865.32641602,\n",
       "        -1864.60229492, -1864.68969727, -1864.5078125 , -1864.14367676,\n",
       "        -1864.66125488, -1864.49707031, -1864.34228516, -1868.7746582 ,\n",
       "        -1866.11975098, -1864.10339355, -1865.88745117, -1865.04406738,\n",
       "        -1864.59179688, -1868.28613281, -1864.68066406, -1865.23620605,\n",
       "        -1864.92553711, -1865.20373535, -1867.05297852, -1865.64587402,\n",
       "        -1865.43798828, -1866.17651367, -1866.71057129, -1868.95495605,\n",
       "        -1865.61315918, -1865.18054199, -1869.65368652, -1869.2689209 ,\n",
       "        -1866.60632324, -1865.73474121, -1866.13830566, -1869.2989502 ]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>tree_depth</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>3 2 3 3 2 3 3 3 ... 3 3 3 3 2 3 3 3</div><input id='attrs-1b6c0f34-7fb0-42cb-aea6-60ff259f83e0' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-1b6c0f34-7fb0-42cb-aea6-60ff259f83e0' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-bac84002-d71b-4be2-802b-b46cd0c1ab94' class='xr-var-data-in' type='checkbox'><label for='data-bac84002-d71b-4be2-802b-b46cd0c1ab94' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[3, 2, 3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 2, 3, 2, 3, 3, 3,\n",
       "        2, 2, 3, 3, 3, 2, 3, 3, 2, 2, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 2,\n",
       "        3, 2, 2, 2, 3, 2, 3, 3, 1, 2, 3, 3, 3, 3, 2, 2, 3, 2, 3, 2, 2, 3,\n",
       "        3, 3, 3, 3, 3, 3, 2, 3, 2, 3, 3, 3, 3, 2, 3, 2, 3, 2, 3, 2, 2, 2,\n",
       "        3, 2, 3, 3, 3, 3, 3, 2, 3, 3, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 2, 2, 3, 3, 2, 3, 3, 3, 3, 2, 3, 3,\n",
       "        2, 3, 3, 3, 3, 2, 3, 3, 3, 2, 3, 3, 2, 3, 3, 2, 2, 3, 3, 3, 3, 3,\n",
       "        3, 2, 2, 2, 3, 2, 3, 2, 3, 2, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3, 2,\n",
       "        3, 3, 3, 3, 3, 3, 2, 3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 3, 3, 2, 3, 3,\n",
       "        3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 2, 2, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 2, 2, 3, 2, 3, 3, 3, 2, 2,\n",
       "        3, 3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3,\n",
       "        3, 2, 3, 3, 3, 2, 2, 3, 2, 2, 3, 3, 3, 1, 2, 2, 3, 3, 3, 3, 2, 3,\n",
       "        3, 3, 3, 3, 2, 2, 3, 3, 3, 2, 2, 2, 3, 3, 3, 2, 3, 3, 3, 2, 3, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 2, 3, 3, 2, 3, 3, 3,\n",
       "        3, 3, 3, 2, 3, 3, 2, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 2, 2, 2, 2, 2, 3, 3, 2, 2, 3, 3, 3, 3, 2, 3, 2, 3, 3, 3, 2,\n",
       "        3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 2, 2, 3, 2, 3, 3, 3, 2, 2, 3, 3, 2,\n",
       "        2, 2, 3, 2, 3, 3, 2, 3, 3, 3, 3, 3, 3, 2, 3, 2, 3, 2, 3, 3, 3, 3,\n",
       "...\n",
       "        2, 3, 3, 2, 3, 3, 2, 2, 3, 3, 2, 3, 3, 3, 2, 2, 2, 3, 3, 2, 2, 3,\n",
       "        2, 3, 3, 2, 3, 3, 1, 3, 2, 3, 2, 2, 3, 3, 3, 2, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 3, 2, 3, 2, 3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 3, 3, 2, 3, 3,\n",
       "        3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 2, 2, 2, 3, 3, 3, 2, 3, 3,\n",
       "        3, 3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 2, 3, 3, 3, 3, 3, 2,\n",
       "        3, 2, 3, 3, 3, 3, 3, 3, 2, 3, 3, 2, 3, 2, 3, 3, 2, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3, 3, 3, 3, 2, 3, 2, 2, 2, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3, 2, 2, 3, 3, 3, 3, 3, 2, 3,\n",
       "        2, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 2, 3, 3, 3, 3, 2, 3, 2, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 2, 3, 3, 3, 2, 3, 2, 3, 3,\n",
       "        3, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 2,\n",
       "        3, 3, 3, 3, 2, 3, 2, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3, 3, 2, 2, 3,\n",
       "        3, 3, 2, 3, 2, 2, 3, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2,\n",
       "        3, 3, 3, 3, 3, 3, 2, 3, 3, 2, 3, 3, 2, 3, 2, 3, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 3, 2, 2, 3, 3, 3, 3, 3, 3, 3, 2, 3,\n",
       "        2, 3, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,\n",
       "        3, 3, 3, 2, 3, 3, 3, 2, 3, 3, 3, 3, 2, 3, 3, 3, 3, 3, 2, 3, 3, 3,\n",
       "        3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 3, 3, 3]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>energy_error</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.1715 -0.09475 ... 0.1364 1.595</div><input id='attrs-e47877ba-79f6-46ca-b580-1cf44793e3b4' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-e47877ba-79f6-46ca-b580-1cf44793e3b4' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-df660c38-0cfa-403d-b96d-d7061b581e83' class='xr-var-data-in' type='checkbox'><label for='data-df660c38-0cfa-403d-b96d-d7061b581e83' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[ 1.71516418e-01, -9.47513580e-02, -2.98373699e-02,\n",
       "         3.28057706e-02, -5.61435223e-02,  1.50852621e-01,\n",
       "         2.51241505e-01, -2.42199898e-01, -3.54073942e-01,\n",
       "         3.73437941e-01,  6.52290583e-02, -9.60764885e-02,\n",
       "         1.21287441e+00, -5.90922832e-02,  8.62588882e-02,\n",
       "        -1.41063929e-01, -1.06447220e+00, -6.53385639e-01,\n",
       "        -3.38342786e-02,  1.50417894e-01, -2.05376148e-01,\n",
       "         7.49939919e-01,  5.03122807e-03, -3.16870451e-01,\n",
       "        -2.86505282e-01, -2.79041290e-01, -4.80747223e-03,\n",
       "        -1.20923519e-01,  3.12452376e-01, -1.76711082e-02,\n",
       "         1.22884274e-01,  3.32394123e-01, -4.23251152e-01,\n",
       "         1.00963116e+00, -1.20626330e+00,  3.44390631e-01,\n",
       "        -2.93804407e-01,  1.23666644e-01,  1.69238091e-01,\n",
       "        -8.99507999e-02, -2.35871077e-01,  1.14394307e-01,\n",
       "         1.55564666e-01, -2.67923832e-01, -3.45338583e-02,\n",
       "         1.36649609e-03, -3.44636440e-02,  2.18291283e-02,\n",
       "        -9.04827118e-02,  1.94438934e-01,  1.14177227e-01,\n",
       "        -1.44228458e-01, -1.47991717e-01,  5.96048117e-01,\n",
       "        -6.40288591e-01, -7.26509094e-03,  2.91541815e-02,\n",
       "        -2.73256302e-02,  5.49527407e-02, -1.48508310e-01,\n",
       "...\n",
       "        -7.21142292e-02,  3.84603977e-01, -1.20925903e-03,\n",
       "        -3.92291546e-01,  1.07084513e-02,  0.00000000e+00,\n",
       "         1.13795519e-01, -1.85956955e-01, -2.18558311e-02,\n",
       "        -7.67141581e-03,  9.82884467e-02,  4.37744856e-02,\n",
       "         1.89290524e-01, -1.96200371e-01,  1.11245513e-01,\n",
       "        -2.13154107e-01,  4.03988361e-02,  8.34615231e-02,\n",
       "         8.96778107e-02, -1.38068557e-01,  4.24627781e-01,\n",
       "        -2.17598438e-01, -1.24223232e-02,  1.61187410e-01,\n",
       "         7.19680786e-02,  3.49573851e-01, -2.79981852e-01,\n",
       "        -5.97476959e-02,  4.51509953e-02, -2.71989107e-02,\n",
       "        -7.37786293e-02,  1.91808820e-01, -2.25873947e-01,\n",
       "        -2.23549455e-02,  1.61478090e+00, -1.00723672e+00,\n",
       "        -2.41398692e-01,  2.26678014e-01, -1.36860609e-01,\n",
       "        -4.38078195e-02,  4.50257659e-01, -5.67580521e-01,\n",
       "         1.93203926e-01, -1.72006130e-01,  1.17813587e-01,\n",
       "        -6.16631508e-02,  3.01395416e-01, -2.47859240e-01,\n",
       "         9.95254517e-02,  3.41851711e-01,  4.39083457e-01,\n",
       "        -1.04033184e+00, -2.60493755e-02,  1.72959983e-01,\n",
       "        -2.40087509e-04,  5.55431843e-03, -2.82466888e-01,\n",
       "         1.36364460e-01,  1.59498310e+00]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>index_in_trajectory</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>-4 -1 -4 5 3 -6 4 ... 3 5 2 1 2 -5</div><input id='attrs-50edf425-3fd6-4834-938b-80c4b35ceea2' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-50edf425-3fd6-4834-938b-80c4b35ceea2' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-314a064e-7457-4ac0-b0b1-f742986cfff8' class='xr-var-data-in' type='checkbox'><label for='data-314a064e-7457-4ac0-b0b1-f742986cfff8' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[-4, -1, -4,  5,  3, -6,  4, -3, -1, -6, -5, -6, -2, -1, -1,  6,\n",
       "        -1, -5, -1, -2, -3,  1,  3, -2, -3,  3,  5, -3, -3, -5,  1,  3,\n",
       "        -2,  2, -3, -4,  3, -7,  4,  2,  3,  4,  1,  3, -4, -3, -3, -3,\n",
       "         3, -1,  6, -3, -1, -2,  1,  6, -2, -6, -1, -3,  1,  3, -2,  0,\n",
       "         3, -3,  2,  3,  5,  2, -1, -1, -3, -2,  2, -2,  2,  1, -4,  3,\n",
       "        -1,  0,  4, -3, -5,  2, -3, -3,  5,  3,  7,  5,  2, -6, -4,  3,\n",
       "        -2, -2,  2,  3,  5, -5, -2, -3,  3, -5, -1, -3, -3, -2, -2,  6,\n",
       "        -5,  1, -4,  3,  7, -3, -3, -1, -3, -1, -3,  3, -5, -4, -1,  2,\n",
       "         1, -4,  4, -3, -4,  1,  4,  5, -2,  2,  1,  1, -2,  1,  3,  1,\n",
       "         7,  4,  1, -2, -2,  1,  1, -2,  5, -4, -2, -1, -1,  3, -3, -3,\n",
       "         3,  3,  1, -2, -4,  1, -2, -1,  2, -2,  1, -2,  2,  5, -6,  5,\n",
       "        -3,  1, -3,  1, -5,  1, -5, -3,  1,  2,  5, -5,  2, -3,  4, -2,\n",
       "         3,  3,  2, -1, -3,  2,  3, -2, -3, -2,  2,  5,  3, -1,  6,  1,\n",
       "         5,  2,  3, -1,  2,  2,  2,  2,  1,  3,  2,  1,  2,  2,  3,  4,\n",
       "         6, -3, -1, -6,  5, -4,  2,  1,  3, -7, -1, -1,  1,  3,  1, -7,\n",
       "         5, -3,  5, -6, -4, -2,  4, -2,  2,  5, -3, -1,  3,  2,  4,  2,\n",
       "        -2,  3,  2,  3, -2,  3, -3, -3, -4,  6, -1, -1,  5, -4, -2, -5,\n",
       "         4,  3, -5, -1,  2,  1,  7, -1, -6, -3, -2,  1, -2, -2,  2, -1,\n",
       "        -3,  1, -4,  0, -1,  5, -2, -2,  0, -2,  1,  1, -3, -2,  3, -2,\n",
       "         5,  6, -3, -5,  2,  1,  5,  4, -1,  1,  3,  3, -3,  1, -1, -2,\n",
       "...\n",
       "         2, -4,  2,  4, -3, -5, -4, -4, -3,  6,  4, -6, -5, -5,  3,  2,\n",
       "        -2, -3, -2, -6, -4, -1,  4,  6,  5, -2, -1, -4,  3,  4,  2, -4,\n",
       "         3, -6, -3,  4, -1,  4,  6,  7, -5,  6, -2, -1,  2, -4, -2,  2,\n",
       "         1,  5, -3,  6, -2,  3, -4,  2, -7,  1,  2, -5, -5, -1,  2,  2,\n",
       "         2, -6, -4,  2, -3, -6,  3, -2, -2,  1, -2,  6,  5, -3,  6, -4,\n",
       "         3, -5,  4, -1, -2, -5,  4, -2,  3,  3,  5, -3,  5,  4,  4,  4,\n",
       "         7,  1, -2, -3,  2,  6, -3,  5, -2, -1,  6, -3, -4,  3, -3,  2,\n",
       "         5,  5, -4, -1, -2, -5, -3, -4, -4, -1,  5, -5, -6, -4,  2,  4,\n",
       "        -5,  2, -2,  6, -1, -1, -5, -3, -6,  2,  1,  1, -2,  4,  5,  1,\n",
       "        -6, -6, -5, -4, -5, -2,  4, -6, -5, -4,  1,  2,  5, -4,  1,  2,\n",
       "         3,  2, -5,  5, -2, -2, -1, -5, -5, -3,  5, -6,  3,  4,  1, -4,\n",
       "         4, -7,  1, -3, -2,  5,  4, -1, -3,  1,  1,  2,  2, -1, -1,  3,\n",
       "        -3, -5,  7,  3, -1,  2, -3,  4, -2,  4, -6, -1, -3,  3,  0,  7,\n",
       "         4,  4,  1,  1, -4,  2,  3,  1, -3,  2, -3,  1, -1,  3,  3,  3,\n",
       "        -2, -1, -3,  4, -2,  4,  3, -1,  2,  3,  1,  5,  2, -1, -2,  2,\n",
       "         5,  5,  5, -1, -3,  5,  1, -6, -1, -6, -2, -5, -1,  3,  0,  5,\n",
       "        -2, -4,  5, -3,  3,  2, -6,  6,  6,  4,  2, -5,  3, -3, -3, -5,\n",
       "         2,  3,  5,  3,  2,  3, -2, -1, -1,  4, -1, -6, -1, -4, -4,  2,\n",
       "         1,  3,  3,  1, -7,  6, -3, -3,  3,  3, -1,  5, -1, -5,  3,  5,\n",
       "         2,  1,  2, -5]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>perf_counter_diff</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.01556 0.008141 ... 0.0157 0.01593</div><input id='attrs-bd930d84-79b1-4e09-86ac-8d5f4a287990' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-bd930d84-79b1-4e09-86ac-8d5f4a287990' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-7292f47a-032d-49ed-907d-e053846c1f94' class='xr-var-data-in' type='checkbox'><label for='data-7292f47a-032d-49ed-907d-e053846c1f94' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.01556438, 0.00814067, 0.01634   , 0.01621371, 0.00784433,\n",
       "        0.01648213, 0.01586158, 0.01589417, 0.00826788, 0.01696092,\n",
       "        0.016115  , 0.01615258, 0.00780104, 0.016276  , 0.01555683,\n",
       "        0.01657979, 0.00826325, 0.01536017, 0.00791967, 0.01601975,\n",
       "        0.01539483, 0.01606563, 0.00839658, 0.00790696, 0.01596254,\n",
       "        0.01580954, 0.02050483, 0.008414  , 0.01619725, 0.01606658,\n",
       "        0.00806504, 0.00795229, 0.01609967, 0.01604475, 0.01560742,\n",
       "        0.01549613, 0.01607667, 0.01649425, 0.01515267, 0.00831283,\n",
       "        0.01588675, 0.01601121, 0.015704  , 0.00781629, 0.01626717,\n",
       "        0.00833783, 0.008159  , 0.00806762, 0.01605758, 0.00793113,\n",
       "        0.01624504, 0.01614975, 0.00407233, 0.00830362, 0.01634512,\n",
       "        0.01583887, 0.01614333, 0.01678779, 0.00854808, 0.00856892,\n",
       "        0.01522383, 0.0079735 , 0.01711271, 0.00813079, 0.0086315 ,\n",
       "        0.01675346, 0.01557096, 0.01601708, 0.01566054, 0.01512063,\n",
       "        0.0156695 , 0.01601721, 0.00807346, 0.01606904, 0.00795033,\n",
       "        0.01532754, 0.01661454, 0.01581275, 0.01597425, 0.00809742,\n",
       "        0.01578933, 0.00772867, 0.01595846, 0.00823617, 0.01642254,\n",
       "        0.00801496, 0.00790854, 0.00750387, 0.015675  , 0.00831121,\n",
       "        0.01651263, 0.01586837, 0.01656771, 0.01569771, 0.01655071,\n",
       "        0.00788558, 0.01657675, 0.01612567, 0.00795221, 0.00809846,\n",
       "...\n",
       "        0.01593208, 0.01541792, 0.00746938, 0.01606775, 0.01619838,\n",
       "        0.00827663, 0.01595413, 0.01622483, 0.00813375, 0.01570112,\n",
       "        0.00782413, 0.0155345 , 0.01611475, 0.01665087, 0.01532388,\n",
       "        0.01514142, 0.01598571, 0.01493983, 0.01563192, 0.01509775,\n",
       "        0.00765825, 0.01517796, 0.01528954, 0.01493013, 0.00814675,\n",
       "        0.01551283, 0.01538546, 0.01520542, 0.01572558, 0.00762104,\n",
       "        0.00766458, 0.01480871, 0.01653346, 0.01603237, 0.01576929,\n",
       "        0.01567154, 0.01641438, 0.02713458, 0.00774483, 0.01564179,\n",
       "        0.00804183, 0.015333  , 0.00783329, 0.01535362, 0.01658083,\n",
       "        0.01568767, 0.01557158, 0.01502533, 0.01578254, 0.01611225,\n",
       "        0.015475  , 0.01553017, 0.01505587, 0.01517387, 0.01574758,\n",
       "        0.0153705 , 0.01534158, 0.01515067, 0.01584808, 0.01515812,\n",
       "        0.01556879, 0.01475642, 0.01532133, 0.01569537, 0.01538917,\n",
       "        0.00773363, 0.01528571, 0.01573983, 0.01548825, 0.00787829,\n",
       "        0.01544342, 0.01541471, 0.01501658, 0.01498   , 0.00780646,\n",
       "        0.01560404, 0.01540408, 0.01518912, 0.01553333, 0.01533129,\n",
       "        0.00786663, 0.01565925, 0.01538838, 0.0151395 , 0.01539812,\n",
       "        0.01470329, 0.01534104, 0.01486267, 0.01592942, 0.01580167,\n",
       "        0.01569829, 0.01557867, 0.01550004, 0.01470879, 0.01504521,\n",
       "        0.01495475, 0.00753967, 0.01550121, 0.01570058, 0.01592725]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>diverging</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>bool</div><div class='xr-var-preview xr-preview'>False False False ... False False</div><input id='attrs-d358a53e-4cf8-40e1-8e2b-5f6746284037' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-d358a53e-4cf8-40e1-8e2b-5f6746284037' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-aee0cb7a-cba5-4c90-a88c-64928bf38f50' class='xr-var-data-in' type='checkbox'><label for='data-aee0cb7a-cba5-4c90-a88c-64928bf38f50' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "...\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False, False, False, False, False,\n",
       "        False, False, False, False, False]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>energy</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>1.867e+03 1.868e+03 ... 1.873e+03</div><input id='attrs-f6cd7218-1aea-499b-b76c-33c2526b65fb' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-f6cd7218-1aea-499b-b76c-33c2526b65fb' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-6c4ac61d-025d-41bf-baa2-6fb02601a4ae' class='xr-var-data-in' type='checkbox'><label for='data-6c4ac61d-025d-41bf-baa2-6fb02601a4ae' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[1866.85063004, 1868.14417696, 1865.10505986, 1865.20036575,\n",
       "        1865.93262839, 1865.49946469, 1866.80378896, 1868.25561619,\n",
       "        1866.43042564, 1866.22382194, 1866.60871732, 1866.90798259,\n",
       "        1870.04307222, 1868.93804169, 1869.88924885, 1866.52074313,\n",
       "        1870.5670805 , 1871.0106585 , 1868.83762681, 1869.78599188,\n",
       "        1869.84417057, 1870.86864424, 1869.50992072, 1868.5024116 ,\n",
       "        1866.91573596, 1869.09869385, 1870.5679512 , 1868.11102939,\n",
       "        1868.14350563, 1874.71006966, 1877.95511484, 1875.24459553,\n",
       "        1869.48305297, 1870.80907679, 1870.1706233 , 1867.64320827,\n",
       "        1867.75180697, 1867.08121562, 1869.54988074, 1869.54081917,\n",
       "        1867.28529596, 1866.33686793, 1867.46773744, 1868.22154188,\n",
       "        1866.75670612, 1866.59367657, 1869.14723635, 1868.09101963,\n",
       "        1867.00981927, 1867.85805798, 1872.317379  , 1870.03810787,\n",
       "        1865.09996164, 1869.09844708, 1869.74869776, 1868.00045919,\n",
       "        1867.2640717 , 1866.64579701, 1867.16698349, 1866.30961728,\n",
       "        1866.70782113, 1866.54416788, 1866.8243711 , 1868.47781062,\n",
       "        1867.87083733, 1872.08012819, 1872.41593361, 1870.64814705,\n",
       "        1869.90035343, 1867.25095177, 1864.63483205, 1867.47706079,\n",
       "        1868.74094129, 1868.80557394, 1866.64381647, 1866.94230378,\n",
       "        1869.78194308, 1870.63659716, 1868.55713916, 1866.67523968,\n",
       "...\n",
       "        1868.22560716, 1867.95629966, 1868.99762845, 1868.63799024,\n",
       "        1867.77707243, 1867.61065483, 1868.71504211, 1869.46704459,\n",
       "        1867.99021137, 1866.78289676, 1866.89354968, 1870.60876036,\n",
       "        1869.31694102, 1869.92723632, 1867.38501668, 1869.01391059,\n",
       "        1870.62121058, 1868.96025634, 1868.93604636, 1869.08163524,\n",
       "        1869.85151172, 1869.03447652, 1870.44969463, 1872.72139835,\n",
       "        1872.91097832, 1866.94426394, 1870.19799852, 1868.81869864,\n",
       "        1868.82480836, 1866.40316415, 1864.96830213, 1865.76459709,\n",
       "        1866.3345778 , 1871.91431475, 1868.59512186, 1866.6277045 ,\n",
       "        1865.52627683, 1868.482476  , 1868.94414711, 1869.50339532,\n",
       "        1868.61560893, 1867.5917871 , 1868.26568866, 1867.84008467,\n",
       "        1867.21255422, 1872.8345232 , 1871.03453517, 1868.76417637,\n",
       "        1866.11830521, 1867.12766147, 1866.56856251, 1867.19766712,\n",
       "        1866.12751853, 1865.29751998, 1864.60789296, 1871.15085649,\n",
       "        1869.19926906, 1867.19939661, 1867.24468005, 1868.7216258 ,\n",
       "        1865.13504845, 1870.01514947, 1868.4832046 , 1870.77413368,\n",
       "        1869.86999702, 1867.96875048, 1869.85020113, 1870.32903385,\n",
       "        1866.30831575, 1867.09134662, 1867.85212243, 1870.2972759 ,\n",
       "        1870.6812396 , 1866.69262695, 1870.31594914, 1872.64465523,\n",
       "        1870.44711924, 1867.28014779, 1869.31700134, 1872.56366491]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>smallest_eigval</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>nan nan nan nan ... nan nan nan nan</div><input id='attrs-91a925ea-6d6e-4914-95bc-b1b449a9f1d5' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-91a925ea-6d6e-4914-95bc-b1b449a9f1d5' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-7a51d620-7b43-4aaa-af9b-3082dac47a83' class='xr-var-data-in' type='checkbox'><label for='data-7a51d620-7b43-4aaa-af9b-3082dac47a83' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "...\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan,\n",
       "        nan, nan, nan, nan, nan, nan]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>acceptance_rate</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.8446 0.8703 ... 0.6986 0.3913</div><input id='attrs-cdebfa4f-caf9-4c63-bda3-6edd23293ec4' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-cdebfa4f-caf9-4c63-bda3-6edd23293ec4' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-263492d3-8ac7-4ed3-b5fb-2b76a356e323' class='xr-var-data-in' type='checkbox'><label for='data-263492d3-8ac7-4ed3-b5fb-2b76a356e323' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.84455343, 0.87029402, 0.96882652, 0.9424318 , 0.82784664,\n",
       "        0.88262425, 0.85166604, 0.89700705, 0.99249693, 0.73702138,\n",
       "        0.959214  , 0.99626821, 0.21748304, 0.64818765, 0.7930789 ,\n",
       "        0.9662217 , 0.88073204, 0.8771015 , 0.99563772, 0.915964  ,\n",
       "        1.        , 0.64018523, 0.99832714, 1.        , 0.99910771,\n",
       "        0.8187805 , 0.92604064, 0.80699519, 0.85199975, 0.4503738 ,\n",
       "        0.6038931 , 0.60812647, 0.96815662, 0.49368127, 0.78268865,\n",
       "        0.75184079, 0.95279852, 0.96723353, 0.8920603 , 0.94583668,\n",
       "        1.        , 0.96670014, 0.88391394, 0.84796014, 0.98823556,\n",
       "        0.95076009, 0.74202094, 0.87914989, 0.8364116 , 0.88975085,\n",
       "        0.88261532, 0.79729772, 1.        , 0.72937672, 0.96992657,\n",
       "        0.96077237, 0.98924829, 0.97571461, 0.85031559, 0.98051321,\n",
       "        0.97574515, 0.88393114, 0.9128673 , 0.28443898, 0.81674136,\n",
       "        0.83031699, 0.874144  , 0.3236842 , 0.91357658, 0.98902236,\n",
       "        0.99589461, 0.79574497, 0.53634238, 0.89920678, 0.95369762,\n",
       "        0.96944832, 0.4625893 , 1.        , 0.8566259 , 0.76871749,\n",
       "        0.93573691, 0.00586145, 0.88060384, 0.7624515 , 0.88862949,\n",
       "        0.90348415, 0.60828967, 0.99760909, 0.95940759, 0.93853534,\n",
       "        0.977193  , 0.91660907, 0.94922943, 0.54740938, 0.96510334,\n",
       "        0.86801316, 0.70780264, 0.87749276, 0.18708535, 0.61316859,\n",
       "...\n",
       "        0.99917725, 0.96576962, 0.92913673, 0.83136183, 0.84742738,\n",
       "        0.9362504 , 0.96875596, 0.91413387, 0.99505419, 0.70284235,\n",
       "        0.8987336 , 0.56454669, 0.95750551, 0.51666756, 0.97891538,\n",
       "        0.96899013, 0.70366156, 0.90617993, 0.5936749 , 0.91311876,\n",
       "        0.79366736, 0.86557476, 0.9578969 , 0.94528351, 0.49740462,\n",
       "        0.81261606, 0.8140037 , 0.86524419, 0.84639623, 0.97865839,\n",
       "        0.9945704 , 0.69405434, 1.        , 0.76313746, 0.99925544,\n",
       "        0.77981212, 0.84500519, 0.52152333, 0.88376542, 1.        ,\n",
       "        0.69967908, 1.        , 0.75479121, 0.97555582, 0.46147716,\n",
       "        0.98802462, 0.45772944, 0.69433142, 0.91193521, 0.94342509,\n",
       "        0.96951828, 0.92637536, 0.98139673, 0.33063639, 0.9881543 ,\n",
       "        0.91100786, 0.99904181, 0.82052456, 0.88840512, 0.95924111,\n",
       "        0.99576466, 0.70622679, 0.98591532, 0.99086539, 0.92237413,\n",
       "        0.38362488, 0.76540242, 0.99177278, 0.92109911, 0.81363271,\n",
       "        0.92190735, 0.86214041, 0.7933622 , 0.95578031, 1.        ,\n",
       "        0.29331821, 0.98943386, 0.99797407, 0.84336505, 0.77985317,\n",
       "        1.        , 0.56096919, 0.62201292, 0.71728082, 0.73298132,\n",
       "        0.8609413 , 0.74290859, 0.51995317, 1.        , 0.97018497,\n",
       "        0.87226905, 0.91546336, 1.        , 0.95947066, 0.89647451,\n",
       "        0.97401367, 0.88817153, 0.97253052, 0.69862563, 0.39133288]])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>process_time_diff</span></div><div class='xr-var-dims'>(chain, draw)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>0.03068 0.01581 ... 0.03003 0.03008</div><input id='attrs-d43a874f-71ad-48fa-94e0-7a717a8b1a7b' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-d43a874f-71ad-48fa-94e0-7a717a8b1a7b' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-67eb5922-922f-4226-819c-4732e8428578' class='xr-var-data-in' type='checkbox'><label for='data-67eb5922-922f-4226-819c-4732e8428578' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.030679, 0.015811, 0.031635, 0.03171 , 0.015667, 0.032007,\n",
       "        0.030938, 0.031335, 0.015929, 0.032808, 0.031171, 0.031386,\n",
       "        0.015385, 0.031707, 0.030808, 0.031642, 0.015947, 0.030772,\n",
       "        0.015673, 0.031162, 0.030576, 0.031794, 0.016414, 0.015581,\n",
       "        0.031141, 0.031474, 0.034605, 0.016219, 0.031844, 0.031907,\n",
       "        0.015465, 0.015692, 0.031464, 0.031059, 0.030655, 0.030326,\n",
       "        0.031838, 0.031641, 0.030884, 0.016181, 0.031048, 0.031292,\n",
       "        0.031124, 0.015561, 0.031517, 0.016344, 0.016305, 0.015514,\n",
       "        0.031291, 0.015547, 0.031951, 0.032247, 0.007847, 0.016158,\n",
       "        0.032322, 0.031346, 0.031451, 0.032425, 0.016064, 0.016595,\n",
       "        0.030584, 0.015791, 0.032547, 0.015752, 0.01649 , 0.032238,\n",
       "        0.03057 , 0.030889, 0.031144, 0.030273, 0.030753, 0.031456,\n",
       "        0.016025, 0.030919, 0.015474, 0.029413, 0.032206, 0.03123 ,\n",
       "        0.031229, 0.015751, 0.031235, 0.01562 , 0.031796, 0.016047,\n",
       "        0.031352, 0.015282, 0.015544, 0.015028, 0.030988, 0.016095,\n",
       "        0.031816, 0.030857, 0.0315  , 0.030626, 0.031573, 0.015161,\n",
       "        0.032308, 0.031128, 0.015499, 0.01546 , 0.031815, 0.030798,\n",
       "        0.031538, 0.030796, 0.031186, 0.031682, 0.03207 , 0.031634,\n",
       "        0.031322, 0.031388, 0.030694, 0.031831, 0.031392, 0.015564,\n",
       "        0.031433, 0.031871, 0.030907, 0.031587, 0.030768, 0.031515,\n",
       "...\n",
       "        0.031801, 0.031606, 0.029935, 0.030467, 0.030132, 0.029168,\n",
       "        0.030583, 0.029435, 0.029502, 0.030166, 0.029734, 0.014835,\n",
       "        0.029635, 0.029585, 0.029383, 0.029305, 0.030588, 0.029444,\n",
       "        0.014552, 0.030732, 0.03084 , 0.015734, 0.030535, 0.03036 ,\n",
       "        0.01536 , 0.030209, 0.014721, 0.029846, 0.030482, 0.031097,\n",
       "        0.030029, 0.029345, 0.030562, 0.029205, 0.029816, 0.029394,\n",
       "        0.01481 , 0.02989 , 0.029537, 0.02909 , 0.015159, 0.03    ,\n",
       "        0.029932, 0.029692, 0.030331, 0.014903, 0.014711, 0.02903 ,\n",
       "        0.031654, 0.030591, 0.030147, 0.030527, 0.03096 , 0.034265,\n",
       "        0.015077, 0.03033 , 0.015369, 0.029652, 0.015273, 0.029498,\n",
       "        0.031045, 0.030202, 0.029862, 0.029235, 0.030147, 0.030571,\n",
       "        0.029877, 0.029919, 0.029314, 0.029325, 0.030404, 0.029532,\n",
       "        0.029831, 0.02928 , 0.029437, 0.029704, 0.030049, 0.028754,\n",
       "        0.029499, 0.030198, 0.029635, 0.014721, 0.029489, 0.030156,\n",
       "        0.029739, 0.01496 , 0.029814, 0.029619, 0.029378, 0.029101,\n",
       "        0.014988, 0.02991 , 0.029944, 0.029391, 0.029768, 0.02955 ,\n",
       "        0.015294, 0.030077, 0.029664, 0.02965 , 0.029658, 0.028906,\n",
       "        0.029585, 0.029154, 0.029965, 0.02991 , 0.029797, 0.029704,\n",
       "        0.029833, 0.029017, 0.029219, 0.029044, 0.014664, 0.029623,\n",
       "        0.030025, 0.030076]])</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-07e3a947-a4de-4d14-9163-24fa83710293' class='xr-section-summary-in' type='checkbox'  ><label for='section-07e3a947-a4de-4d14-9163-24fa83710293' class='xr-section-summary' >Indexes: <span>(2)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-index-name'><div>chain</div></div><div class='xr-index-preview'>PandasIndex</div><div></div><input id='index-c4c154b9-a479-4cb6-968d-3ded7720c7ff' class='xr-index-data-in' type='checkbox'/><label for='index-c4c154b9-a479-4cb6-968d-3ded7720c7ff' title='Show/Hide index repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-index-data'><pre>PandasIndex(Index([0, 1], dtype=&#x27;int64&#x27;, name=&#x27;chain&#x27;))</pre></div></li><li class='xr-var-item'><div class='xr-index-name'><div>draw</div></div><div class='xr-index-preview'>PandasIndex</div><div></div><input id='index-38df1df8-b794-4777-ae4c-c1e41544d59c' class='xr-index-data-in' type='checkbox'/><label for='index-38df1df8-b794-4777-ae4c-c1e41544d59c' title='Show/Hide index repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-index-data'><pre>PandasIndex(Index([  0,   1,   2,   3,   4,   5,   6,   7,   8,   9,\n",
       "       ...\n",
       "       490, 491, 492, 493, 494, 495, 496, 497, 498, 499],\n",
       "      dtype=&#x27;int64&#x27;, name=&#x27;draw&#x27;, length=500))</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-6a40bd88-449d-427b-a1f4-0addcf9344a5' class='xr-section-summary-in' type='checkbox'  checked><label for='section-6a40bd88-449d-427b-a1f4-0addcf9344a5' class='xr-section-summary' >Attributes: <span>(8)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><dl class='xr-attrs'><dt><span>created_at :</span></dt><dd>2023-06-22T18:12:24.942485</dd><dt><span>arviz_version :</span></dt><dd>0.14.0</dd><dt><span>inference_library :</span></dt><dd>pymc</dd><dt><span>inference_library_version :</span></dt><dd>5.5.0</dd><dt><span>sampling_time :</span></dt><dd>35.316415786743164</dd><dt><span>tuning_steps :</span></dt><dd>500</dd><dt><span>modeling_interface :</span></dt><dd>bambi</dd><dt><span>modeling_interface_version :</span></dt><dd>0.11.0</dd></dl></div></li></ul></div></div><br></div>\n",
       "                      </ul>\n",
       "                  </div>\n",
       "            </li>\n",
       "            \n",
       "            <li class = \"xr-section-item\">\n",
       "                  <input id=\"idata_observed_data5a8ecb79-5e2b-4877-bbca-17acbaa09beb\" class=\"xr-section-summary-in\" type=\"checkbox\">\n",
       "                  <label for=\"idata_observed_data5a8ecb79-5e2b-4877-bbca-17acbaa09beb\" class = \"xr-section-summary\">observed_data</label>\n",
       "                  <div class=\"xr-section-inline-details\"></div>\n",
       "                  <div class=\"xr-section-details\">\n",
       "                      <ul id=\"xr-dataset-coord-list\" class=\"xr-var-list\">\n",
       "                          <div style=\"padding-left:2rem;\"><div><svg style=\"position: absolute; width: 0; height: 0; overflow: hidden\">\n",
       "<defs>\n",
       "<symbol id=\"icon-database\" viewBox=\"0 0 32 32\">\n",
       "<path d=\"M16 0c-8.837 0-16 2.239-16 5v4c0 2.761 7.163 5 16 5s16-2.239 16-5v-4c0-2.761-7.163-5-16-5z\"></path>\n",
       "<path d=\"M16 17c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n",
       "<path d=\"M16 26c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n",
       "</symbol>\n",
       "<symbol id=\"icon-file-text2\" viewBox=\"0 0 32 32\">\n",
       "<path d=\"M28.681 7.159c-0.694-0.947-1.662-2.053-2.724-3.116s-2.169-2.030-3.116-2.724c-1.612-1.182-2.393-1.319-2.841-1.319h-15.5c-1.378 0-2.5 1.121-2.5 2.5v27c0 1.378 1.122 2.5 2.5 2.5h23c1.378 0 2.5-1.122 2.5-2.5v-19.5c0-0.448-0.137-1.23-1.319-2.841zM24.543 5.457c0.959 0.959 1.712 1.825 2.268 2.543h-4.811v-4.811c0.718 0.556 1.584 1.309 2.543 2.268zM28 29.5c0 0.271-0.229 0.5-0.5 0.5h-23c-0.271 0-0.5-0.229-0.5-0.5v-27c0-0.271 0.229-0.5 0.5-0.5 0 0 15.499-0 15.5 0v7c0 0.552 0.448 1 1 1h7v19.5z\"></path>\n",
       "<path d=\"M23 26h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "<path d=\"M23 22h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "<path d=\"M23 18h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n",
       "</symbol>\n",
       "</defs>\n",
       "</svg>\n",
       "<style>/* CSS stylesheet for displaying xarray objects in jupyterlab.\n",
       " *\n",
       " */\n",
       "\n",
       ":root {\n",
       "  --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));\n",
       "  --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));\n",
       "  --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));\n",
       "  --xr-border-color: var(--jp-border-color2, #e0e0e0);\n",
       "  --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);\n",
       "  --xr-background-color: var(--jp-layout-color0, white);\n",
       "  --xr-background-color-row-even: var(--jp-layout-color1, white);\n",
       "  --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);\n",
       "}\n",
       "\n",
       "html[theme=dark],\n",
       "body[data-theme=dark],\n",
       "body.vscode-dark {\n",
       "  --xr-font-color0: rgba(255, 255, 255, 1);\n",
       "  --xr-font-color2: rgba(255, 255, 255, 0.54);\n",
       "  --xr-font-color3: rgba(255, 255, 255, 0.38);\n",
       "  --xr-border-color: #1F1F1F;\n",
       "  --xr-disabled-color: #515151;\n",
       "  --xr-background-color: #111111;\n",
       "  --xr-background-color-row-even: #111111;\n",
       "  --xr-background-color-row-odd: #313131;\n",
       "}\n",
       "\n",
       ".xr-wrap {\n",
       "  display: block !important;\n",
       "  min-width: 300px;\n",
       "  max-width: 700px;\n",
       "}\n",
       "\n",
       ".xr-text-repr-fallback {\n",
       "  /* fallback to plain text repr when CSS is not injected (untrusted notebook) */\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-header {\n",
       "  padding-top: 6px;\n",
       "  padding-bottom: 6px;\n",
       "  margin-bottom: 4px;\n",
       "  border-bottom: solid 1px var(--xr-border-color);\n",
       "}\n",
       "\n",
       ".xr-header > div,\n",
       ".xr-header > ul {\n",
       "  display: inline;\n",
       "  margin-top: 0;\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-obj-type,\n",
       ".xr-array-name {\n",
       "  margin-left: 2px;\n",
       "  margin-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-obj-type {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-sections {\n",
       "  padding-left: 0 !important;\n",
       "  display: grid;\n",
       "  grid-template-columns: 150px auto auto 1fr 20px 20px;\n",
       "}\n",
       "\n",
       ".xr-section-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-section-item input {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-item input + label {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label {\n",
       "  cursor: pointer;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label:hover {\n",
       "  color: var(--xr-font-color0);\n",
       "}\n",
       "\n",
       ".xr-section-summary {\n",
       "  grid-column: 1;\n",
       "  color: var(--xr-font-color2);\n",
       "  font-weight: 500;\n",
       "}\n",
       "\n",
       ".xr-section-summary > span {\n",
       "  display: inline-block;\n",
       "  padding-left: 0.5em;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in + label:before {\n",
       "  display: inline-block;\n",
       "  content: '►';\n",
       "  font-size: 11px;\n",
       "  width: 15px;\n",
       "  text-align: center;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label:before {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label:before {\n",
       "  content: '▼';\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label > span {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-summary,\n",
       ".xr-section-inline-details {\n",
       "  padding-top: 4px;\n",
       "  padding-bottom: 4px;\n",
       "}\n",
       "\n",
       ".xr-section-inline-details {\n",
       "  grid-column: 2 / -1;\n",
       "}\n",
       "\n",
       ".xr-section-details {\n",
       "  display: none;\n",
       "  grid-column: 1 / -1;\n",
       "  margin-bottom: 5px;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked ~ .xr-section-details {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-array-wrap {\n",
       "  grid-column: 1 / -1;\n",
       "  display: grid;\n",
       "  grid-template-columns: 20px auto;\n",
       "}\n",
       "\n",
       ".xr-array-wrap > label {\n",
       "  grid-column: 1;\n",
       "  vertical-align: top;\n",
       "}\n",
       "\n",
       ".xr-preview {\n",
       "  color: var(--xr-font-color3);\n",
       "}\n",
       "\n",
       ".xr-array-preview,\n",
       ".xr-array-data {\n",
       "  padding: 0 5px !important;\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-array-data,\n",
       ".xr-array-in:checked ~ .xr-array-preview {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-array-in:checked ~ .xr-array-data,\n",
       ".xr-array-preview {\n",
       "  display: inline-block;\n",
       "}\n",
       "\n",
       ".xr-dim-list {\n",
       "  display: inline-block !important;\n",
       "  list-style: none;\n",
       "  padding: 0 !important;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list li {\n",
       "  display: inline-block;\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list:before {\n",
       "  content: '(';\n",
       "}\n",
       "\n",
       ".xr-dim-list:after {\n",
       "  content: ')';\n",
       "}\n",
       "\n",
       ".xr-dim-list li:not(:last-child):after {\n",
       "  content: ',';\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-has-index {\n",
       "  font-weight: bold;\n",
       "}\n",
       "\n",
       ".xr-var-list,\n",
       ".xr-var-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-var-item > div,\n",
       ".xr-var-item label,\n",
       ".xr-var-item > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-even);\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-var-item > .xr-var-name:hover span {\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-var-list > li:nth-child(odd) > div,\n",
       ".xr-var-list > li:nth-child(odd) > label,\n",
       ".xr-var-list > li:nth-child(odd) > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-odd);\n",
       "}\n",
       "\n",
       ".xr-var-name {\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-var-dims {\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-var-dtype {\n",
       "  grid-column: 3;\n",
       "  text-align: right;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-preview {\n",
       "  grid-column: 4;\n",
       "}\n",
       "\n",
       ".xr-index-preview {\n",
       "  grid-column: 2 / 5;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-name,\n",
       ".xr-var-dims,\n",
       ".xr-var-dtype,\n",
       ".xr-preview,\n",
       ".xr-attrs dt {\n",
       "  white-space: nowrap;\n",
       "  overflow: hidden;\n",
       "  text-overflow: ellipsis;\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-var-name:hover,\n",
       ".xr-var-dims:hover,\n",
       ".xr-var-dtype:hover,\n",
       ".xr-attrs dt:hover {\n",
       "  overflow: visible;\n",
       "  width: auto;\n",
       "  z-index: 1;\n",
       "}\n",
       "\n",
       ".xr-var-attrs,\n",
       ".xr-var-data,\n",
       ".xr-index-data {\n",
       "  display: none;\n",
       "  background-color: var(--xr-background-color) !important;\n",
       "  padding-bottom: 5px !important;\n",
       "}\n",
       "\n",
       ".xr-var-attrs-in:checked ~ .xr-var-attrs,\n",
       ".xr-var-data-in:checked ~ .xr-var-data,\n",
       ".xr-index-data-in:checked ~ .xr-index-data {\n",
       "  display: block;\n",
       "}\n",
       "\n",
       ".xr-var-data > table {\n",
       "  float: right;\n",
       "}\n",
       "\n",
       ".xr-var-name span,\n",
       ".xr-var-data,\n",
       ".xr-index-name div,\n",
       ".xr-index-data,\n",
       ".xr-attrs {\n",
       "  padding-left: 25px !important;\n",
       "}\n",
       "\n",
       ".xr-attrs,\n",
       ".xr-var-attrs,\n",
       ".xr-var-data,\n",
       ".xr-index-data {\n",
       "  grid-column: 1 / -1;\n",
       "}\n",
       "\n",
       "dl.xr-attrs {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  display: grid;\n",
       "  grid-template-columns: 125px auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt,\n",
       ".xr-attrs dd {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  float: left;\n",
       "  padding-right: 10px;\n",
       "  width: auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt {\n",
       "  font-weight: normal;\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-attrs dt:hover span {\n",
       "  display: inline-block;\n",
       "  background: var(--xr-background-color);\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-attrs dd {\n",
       "  grid-column: 2;\n",
       "  white-space: pre-wrap;\n",
       "  word-break: break-all;\n",
       "}\n",
       "\n",
       ".xr-icon-database,\n",
       ".xr-icon-file-text2,\n",
       ".xr-no-icon {\n",
       "  display: inline-block;\n",
       "  vertical-align: middle;\n",
       "  width: 1em;\n",
       "  height: 1.5em !important;\n",
       "  stroke-width: 0;\n",
       "  stroke: currentColor;\n",
       "  fill: currentColor;\n",
       "}\n",
       "</style><pre class='xr-text-repr-fallback'>&lt;xarray.Dataset&gt;\n",
       "Dimensions:                  (rt,response_obs: 1000, rt,response_extra_dim_0: 2)\n",
       "Coordinates:\n",
       "  * rt,response_obs          (rt,response_obs) int64 0 1 2 3 ... 996 997 998 999\n",
       "  * rt,response_extra_dim_0  (rt,response_extra_dim_0) int64 0 1\n",
       "Data variables:\n",
       "    rt,response              (rt,response_obs, rt,response_extra_dim_0) float32 ...\n",
       "Attributes:\n",
       "    created_at:                  2023-06-22T18:12:24.944352\n",
       "    arviz_version:               0.14.0\n",
       "    inference_library:           pymc\n",
       "    inference_library_version:   5.5.0\n",
       "    modeling_interface:          bambi\n",
       "    modeling_interface_version:  0.11.0</pre><div class='xr-wrap' style='display:none'><div class='xr-header'><div class='xr-obj-type'>xarray.Dataset</div></div><ul class='xr-sections'><li class='xr-section-item'><input id='section-807962d1-889a-458b-a737-900ab60b6919' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-807962d1-889a-458b-a737-900ab60b6919' class='xr-section-summary'  title='Expand/collapse section'>Dimensions:</label><div class='xr-section-inline-details'><ul class='xr-dim-list'><li><span class='xr-has-index'>rt,response_obs</span>: 1000</li><li><span class='xr-has-index'>rt,response_extra_dim_0</span>: 2</li></ul></div><div class='xr-section-details'></div></li><li class='xr-section-item'><input id='section-36c054cd-3aec-4d0d-974f-03d05a52a299' class='xr-section-summary-in' type='checkbox'  checked><label for='section-36c054cd-3aec-4d0d-974f-03d05a52a299' class='xr-section-summary' >Coordinates: <span>(2)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span class='xr-has-index'>rt,response_obs</span></div><div class='xr-var-dims'>(rt,response_obs)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>0 1 2 3 4 5 ... 995 996 997 998 999</div><input id='attrs-03ed7bca-0f91-40bb-8ed0-2a02de5d4df7' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-03ed7bca-0f91-40bb-8ed0-2a02de5d4df7' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-5bb80be2-13af-452b-a491-8b92cb04d104' class='xr-var-data-in' type='checkbox'><label for='data-5bb80be2-13af-452b-a491-8b92cb04d104' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([  0,   1,   2, ..., 997, 998, 999])</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span class='xr-has-index'>rt,response_extra_dim_0</span></div><div class='xr-var-dims'>(rt,response_extra_dim_0)</div><div class='xr-var-dtype'>int64</div><div class='xr-var-preview xr-preview'>0 1</div><input id='attrs-7508c693-99ce-45b4-8115-91cfa5f24c6b' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-7508c693-99ce-45b4-8115-91cfa5f24c6b' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-7ee2bb7d-18e4-4b83-91d1-c8ccf9d164f3' class='xr-var-data-in' type='checkbox'><label for='data-7ee2bb7d-18e4-4b83-91d1-c8ccf9d164f3' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([0, 1])</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-8603e8e6-e81a-40cf-97e3-28c03bf1fd2a' class='xr-section-summary-in' type='checkbox'  checked><label for='section-8603e8e6-e81a-40cf-97e3-28c03bf1fd2a' class='xr-section-summary' >Data variables: <span>(1)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span>rt,response</span></div><div class='xr-var-dims'>(rt,response_obs, rt,response_extra_dim_0)</div><div class='xr-var-dtype'>float32</div><div class='xr-var-preview xr-preview'>0.367 1.0 1.431 ... 1.0 2.203 1.0</div><input id='attrs-5a2fb5e0-787f-4e9c-b588-a490b435a079' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-5a2fb5e0-787f-4e9c-b588-a490b435a079' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-ecc60bcf-0014-4ba9-9b18-76a879ba841b' class='xr-var-data-in' type='checkbox'><label for='data-ecc60bcf-0014-4ba9-9b18-76a879ba841b' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array([[0.36699998, 1.        ],\n",
       "       [1.4309969 , 1.        ],\n",
       "       [0.42400008, 1.        ],\n",
       "       ...,\n",
       "       [1.201992  , 1.        ],\n",
       "       [4.502878  , 1.        ],\n",
       "       [2.203033  , 1.        ]], dtype=float32)</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-0eef7fcb-0060-4d5b-b0a9-2ad4d9c363ee' class='xr-section-summary-in' type='checkbox'  ><label for='section-0eef7fcb-0060-4d5b-b0a9-2ad4d9c363ee' class='xr-section-summary' >Indexes: <span>(2)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-index-name'><div>rt,response_obs</div></div><div class='xr-index-preview'>PandasIndex</div><div></div><input id='index-499fc788-f57b-44e1-a64f-71567bcec538' class='xr-index-data-in' type='checkbox'/><label for='index-499fc788-f57b-44e1-a64f-71567bcec538' title='Show/Hide index repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-index-data'><pre>PandasIndex(Index([  0,   1,   2,   3,   4,   5,   6,   7,   8,   9,\n",
       "       ...\n",
       "       990, 991, 992, 993, 994, 995, 996, 997, 998, 999],\n",
       "      dtype=&#x27;int64&#x27;, name=&#x27;rt,response_obs&#x27;, length=1000))</pre></div></li><li class='xr-var-item'><div class='xr-index-name'><div>rt,response_extra_dim_0</div></div><div class='xr-index-preview'>PandasIndex</div><div></div><input id='index-23b2f935-0cd6-4f73-b42d-c98c5eb8505f' class='xr-index-data-in' type='checkbox'/><label for='index-23b2f935-0cd6-4f73-b42d-c98c5eb8505f' title='Show/Hide index repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-index-data'><pre>PandasIndex(Index([0, 1], dtype=&#x27;int64&#x27;, name=&#x27;rt,response_extra_dim_0&#x27;))</pre></div></li></ul></div></li><li class='xr-section-item'><input id='section-d5fb99b0-c05a-4af7-a527-488e7b3d485a' class='xr-section-summary-in' type='checkbox'  checked><label for='section-d5fb99b0-c05a-4af7-a527-488e7b3d485a' class='xr-section-summary' >Attributes: <span>(6)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><dl class='xr-attrs'><dt><span>created_at :</span></dt><dd>2023-06-22T18:12:24.944352</dd><dt><span>arviz_version :</span></dt><dd>0.14.0</dd><dt><span>inference_library :</span></dt><dd>pymc</dd><dt><span>inference_library_version :</span></dt><dd>5.5.0</dd><dt><span>modeling_interface :</span></dt><dd>bambi</dd><dt><span>modeling_interface_version :</span></dt><dd>0.11.0</dd></dl></div></li></ul></div></div><br></div>\n",
       "                      </ul>\n",
       "                  </div>\n",
       "            </li>\n",
       "            \n",
       "              </ul>\n",
       "            </div>\n",
       "            <style> /* CSS stylesheet for displaying InferenceData objects in jupyterlab.\n",
       " *\n",
       " */\n",
       "\n",
       ":root {\n",
       "  --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));\n",
       "  --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));\n",
       "  --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));\n",
       "  --xr-border-color: var(--jp-border-color2, #e0e0e0);\n",
       "  --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);\n",
       "  --xr-background-color: var(--jp-layout-color0, white);\n",
       "  --xr-background-color-row-even: var(--jp-layout-color1, white);\n",
       "  --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);\n",
       "}\n",
       "\n",
       "html[theme=dark],\n",
       "body.vscode-dark {\n",
       "  --xr-font-color0: rgba(255, 255, 255, 1);\n",
       "  --xr-font-color2: rgba(255, 255, 255, 0.54);\n",
       "  --xr-font-color3: rgba(255, 255, 255, 0.38);\n",
       "  --xr-border-color: #1F1F1F;\n",
       "  --xr-disabled-color: #515151;\n",
       "  --xr-background-color: #111111;\n",
       "  --xr-background-color-row-even: #111111;\n",
       "  --xr-background-color-row-odd: #313131;\n",
       "}\n",
       "\n",
       ".xr-wrap {\n",
       "  display: block;\n",
       "  min-width: 300px;\n",
       "  max-width: 700px;\n",
       "}\n",
       "\n",
       ".xr-text-repr-fallback {\n",
       "  /* fallback to plain text repr when CSS is not injected (untrusted notebook) */\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-header {\n",
       "  padding-top: 6px;\n",
       "  padding-bottom: 6px;\n",
       "  margin-bottom: 4px;\n",
       "  border-bottom: solid 1px var(--xr-border-color);\n",
       "}\n",
       "\n",
       ".xr-header > div,\n",
       ".xr-header > ul {\n",
       "  display: inline;\n",
       "  margin-top: 0;\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-obj-type,\n",
       ".xr-array-name {\n",
       "  margin-left: 2px;\n",
       "  margin-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-obj-type {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-sections {\n",
       "  padding-left: 0 !important;\n",
       "  display: grid;\n",
       "  grid-template-columns: 150px auto auto 1fr 20px 20px;\n",
       "}\n",
       "\n",
       ".xr-sections.group-sections {\n",
       "  grid-template-columns: auto;\n",
       "}\n",
       "\n",
       ".xr-section-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-section-item input {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-item input + label {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label {\n",
       "  cursor: pointer;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-item input:enabled + label:hover {\n",
       "  color: var(--xr-font-color0);\n",
       "}\n",
       "\n",
       ".xr-section-summary {\n",
       "  grid-column: 1;\n",
       "  color: var(--xr-font-color2);\n",
       "  font-weight: 500;\n",
       "}\n",
       "\n",
       ".xr-section-summary > span {\n",
       "  display: inline-block;\n",
       "  padding-left: 0.5em;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label {\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in + label:before {\n",
       "  display: inline-block;\n",
       "  content: '►';\n",
       "  font-size: 11px;\n",
       "  width: 15px;\n",
       "  text-align: center;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:disabled + label:before {\n",
       "  color: var(--xr-disabled-color);\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label:before {\n",
       "  content: '▼';\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked + label > span {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-section-summary,\n",
       ".xr-section-inline-details {\n",
       "  padding-top: 4px;\n",
       "  padding-bottom: 4px;\n",
       "}\n",
       "\n",
       ".xr-section-inline-details {\n",
       "  grid-column: 2 / -1;\n",
       "}\n",
       "\n",
       ".xr-section-details {\n",
       "  display: none;\n",
       "  grid-column: 1 / -1;\n",
       "  margin-bottom: 5px;\n",
       "}\n",
       "\n",
       ".xr-section-summary-in:checked ~ .xr-section-details {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-array-wrap {\n",
       "  grid-column: 1 / -1;\n",
       "  display: grid;\n",
       "  grid-template-columns: 20px auto;\n",
       "}\n",
       "\n",
       ".xr-array-wrap > label {\n",
       "  grid-column: 1;\n",
       "  vertical-align: top;\n",
       "}\n",
       "\n",
       ".xr-preview {\n",
       "  color: var(--xr-font-color3);\n",
       "}\n",
       "\n",
       ".xr-array-preview,\n",
       ".xr-array-data {\n",
       "  padding: 0 5px !important;\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-array-data,\n",
       ".xr-array-in:checked ~ .xr-array-preview {\n",
       "  display: none;\n",
       "}\n",
       "\n",
       ".xr-array-in:checked ~ .xr-array-data,\n",
       ".xr-array-preview {\n",
       "  display: inline-block;\n",
       "}\n",
       "\n",
       ".xr-dim-list {\n",
       "  display: inline-block !important;\n",
       "  list-style: none;\n",
       "  padding: 0 !important;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list li {\n",
       "  display: inline-block;\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "}\n",
       "\n",
       ".xr-dim-list:before {\n",
       "  content: '(';\n",
       "}\n",
       "\n",
       ".xr-dim-list:after {\n",
       "  content: ')';\n",
       "}\n",
       "\n",
       ".xr-dim-list li:not(:last-child):after {\n",
       "  content: ',';\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-has-index {\n",
       "  font-weight: bold;\n",
       "}\n",
       "\n",
       ".xr-var-list,\n",
       ".xr-var-item {\n",
       "  display: contents;\n",
       "}\n",
       "\n",
       ".xr-var-item > div,\n",
       ".xr-var-item label,\n",
       ".xr-var-item > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-even);\n",
       "  margin-bottom: 0;\n",
       "}\n",
       "\n",
       ".xr-var-item > .xr-var-name:hover span {\n",
       "  padding-right: 5px;\n",
       "}\n",
       "\n",
       ".xr-var-list > li:nth-child(odd) > div,\n",
       ".xr-var-list > li:nth-child(odd) > label,\n",
       ".xr-var-list > li:nth-child(odd) > .xr-var-name span {\n",
       "  background-color: var(--xr-background-color-row-odd);\n",
       "}\n",
       "\n",
       ".xr-var-name {\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-var-dims {\n",
       "  grid-column: 2;\n",
       "}\n",
       "\n",
       ".xr-var-dtype {\n",
       "  grid-column: 3;\n",
       "  text-align: right;\n",
       "  color: var(--xr-font-color2);\n",
       "}\n",
       "\n",
       ".xr-var-preview {\n",
       "  grid-column: 4;\n",
       "}\n",
       "\n",
       ".xr-var-name,\n",
       ".xr-var-dims,\n",
       ".xr-var-dtype,\n",
       ".xr-preview,\n",
       ".xr-attrs dt {\n",
       "  white-space: nowrap;\n",
       "  overflow: hidden;\n",
       "  text-overflow: ellipsis;\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-var-name:hover,\n",
       ".xr-var-dims:hover,\n",
       ".xr-var-dtype:hover,\n",
       ".xr-attrs dt:hover {\n",
       "  overflow: visible;\n",
       "  width: auto;\n",
       "  z-index: 1;\n",
       "}\n",
       "\n",
       ".xr-var-attrs,\n",
       ".xr-var-data {\n",
       "  display: none;\n",
       "  background-color: var(--xr-background-color) !important;\n",
       "  padding-bottom: 5px !important;\n",
       "}\n",
       "\n",
       ".xr-var-attrs-in:checked ~ .xr-var-attrs,\n",
       ".xr-var-data-in:checked ~ .xr-var-data {\n",
       "  display: block;\n",
       "}\n",
       "\n",
       ".xr-var-data > table {\n",
       "  float: right;\n",
       "}\n",
       "\n",
       ".xr-var-name span,\n",
       ".xr-var-data,\n",
       ".xr-attrs {\n",
       "  padding-left: 25px !important;\n",
       "}\n",
       "\n",
       ".xr-attrs,\n",
       ".xr-var-attrs,\n",
       ".xr-var-data {\n",
       "  grid-column: 1 / -1;\n",
       "}\n",
       "\n",
       "dl.xr-attrs {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  display: grid;\n",
       "  grid-template-columns: 125px auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt, dd {\n",
       "  padding: 0;\n",
       "  margin: 0;\n",
       "  float: left;\n",
       "  padding-right: 10px;\n",
       "  width: auto;\n",
       "}\n",
       "\n",
       ".xr-attrs dt {\n",
       "  font-weight: normal;\n",
       "  grid-column: 1;\n",
       "}\n",
       "\n",
       ".xr-attrs dt:hover span {\n",
       "  display: inline-block;\n",
       "  background: var(--xr-background-color);\n",
       "  padding-right: 10px;\n",
       "}\n",
       "\n",
       ".xr-attrs dd {\n",
       "  grid-column: 2;\n",
       "  white-space: pre-wrap;\n",
       "  word-break: break-all;\n",
       "}\n",
       "\n",
       ".xr-icon-database,\n",
       ".xr-icon-file-text2 {\n",
       "  display: inline-block;\n",
       "  vertical-align: middle;\n",
       "  width: 1em;\n",
       "  height: 1.5em !important;\n",
       "  stroke-width: 0;\n",
       "  stroke: currentColor;\n",
       "  fill: currentColor;\n",
       "}\n",
       ".xr-wrap{width:700px!important;} </style>"
      ],
      "text/plain": [
       "Inference data with groups:\n",
       "\t> posterior\n",
       "\t> sample_stats\n",
       "\t> observed_data"
      ]
     },
     "execution_count": 39,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_hssm.sample(cores=2, draws=500, tune=500, mp_ctx=\"forkserver\")"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Uploading ONNX Files to a Hugging Face Repository\n",
    "If your ONNX file is not currently housed in your Hugging Face repository, you can include it by adhering to the steps delineated below:"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "1. Import the HfApi module from huggingface_hub:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "from huggingface_hub import HfApi"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "2. Upload the ONNX file using the upload_file method:"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "```python\n",
    "api = HfApi()\n",
    "api.upload_file(\n",
    "    path_or_fileobj=\"test.onnx\",\n",
    "    path_in_repo=\"test.onnx\",\n",
    "    repo_id=\"franklab/HSSM\",\n",
    "    repo_type=\"model\",\n",
    "    create_pr=True,\n",
    ")\n",
    "```"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "The execution of these steps will generate a Pull Request (PR) on Hugging Face, which will subsequently be evaluated by a member of our team."
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Creating a Pull Request and a New ONNX Model\n",
    "\n",
    "1. **Creating a Pull Request on Hugging Face**\n",
    "\n",
    "   Navigate to the following link: [Hugging Face PR](https://huggingface.co/franklab/HSSM/blob/refs%2Fpr%2F1/test.onnx)\n",
    "\n",
    "   By doing so, you will **generate a Pull Request on Hugging Face**, which will be reviewed by our team members.\n",
    "\n",
    "2. **Creating a Custom ONNX Model**\n",
    "\n",
    "   ### Establish a Network Config and State Dictionary Files in PyTorch\n",
    "\n",
    "   To construct a custom model and save it as an ONNX file, you must create a network configuration file and a state dictionary file in PyTorch. Refer to the instructions outlined in the README of the [LANFactory package](LINK_TO_LANFACTORY_PACKAGE).\n",
    "\n",
    "   ### Convert Network Config and State Dictionary Files to ONNX\n",
    "\n",
    "   Once you've generated the network configuration and state dictionary files, you will need to **convert these files into an ONNX format**.\n"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "![onnx conversion](../images/onnx.png)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}
